Change name

Signed-off-by: Rafael Vasquez <[email protected]>

Author: rafvasq

tests/e2e/test_spyre_cb.py

@@ -188,7 +188,7 @@ def test_long_context_batches(
     max_tokens = 10
 
     # (batch_size, prompt_length) pairs
-    batch_prompt_pairs = [
+    batch_token_pairs = [
         (32, 512),
         (16, 1500),
         (8, 3000),
@@ -215,8 +215,8 @@ def test_long_context_batches(
         logprobs=0,
     )
 
-    for batch_size, prompt_len in batch_prompt_pairs:
-        prompt = create_seq_prompt(model, min_token_length=prompt_len)
+    for batch_size, token_len in batch_token_pairs:
+        prompt = create_seq_prompt(model, token_length=token_len)
         prompts = [prompt] * batch_size
 
         vllm_outputs = vllm_model.generate(prompts, sampling_params)

tests/e2e/test_spyre_static_batching_limits.py

@@ -38,9 +38,9 @@ def test_max_prompt_len_and_new_tokens(model: str,
     # Craft a request with a prompt that is slightly too long for the warmup
     # shape
     prompt = create_text_prompt(model,
-                                min_tokens=max_prompt_length,
-                                max_tokens=max_prompt_length + max_new_tokens -
-                                1)
+                                min_token_length=max_prompt_length,
+                                max_token_length=max_prompt_length +
+                                max_new_tokens - 1)
     sampling_params = SamplingParams(max_tokens=1)
 
     with pytest.raises(ValueError, match="warmup"):