iterate over list instead of indeces

Co-authored-by: Sophie du Couédic <[email protected]>
Signed-off-by: Yannick Schnider <[email protected]>

Author: yannicks1

tests/e2e/test_spyre_basic.py

@@ -163,11 +163,11 @@ def test_batch_handling(model: str, backend: str, cb: int,
     max_new_tokens = [5, 20, 10, 5]
 
     vllm_sampling_params = [
-        SamplingParams(max_tokens=max_new_tokens[i],
-                       min_tokens=max_new_tokens[i],
+        SamplingParams(max_tokens=max_tokens_i,
+                       min_tokens=max_tokens_i,
                        temperature=0,
                        ignore_eos=True,
-                       logprobs=0) for i in range(len(max_new_tokens))
+                       logprobs=0) for max_tokens_i in max_new_tokens)
     ]
 
     kwargs = {