Skip to content

Commit 39335d9

Browse files
committed
Fix static test
Signed-off-by: Rafael Vasquez <[email protected]>
1 parent 1f3fc5e commit 39335d9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/e2e/test_spyre_static_batching_limits.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -38,8 +38,8 @@ def test_max_prompt_len_and_new_tokens(model: str,
3838
# Craft a request with a prompt that is slightly too long for the warmup
3939
# shape
4040
prompt = create_text_prompt(model,
41-
min_tokens=max_prompt_length,
42-
max_tokens=max_prompt_length + max_new_tokens -
41+
min_token_length=max_prompt_length,
42+
max_token_length=max_prompt_length + max_new_tokens -
4343
1)
4444
sampling_params = SamplingParams(max_tokens=1)
4545

0 commit comments

Comments
 (0)