Skip to content

Commit efdc43b

Browse files
committed
Lints
Signed-off-by: Rafael Vasquez <[email protected]>
1 parent 39335d9 commit efdc43b

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
@@ -39,8 +39,8 @@ def test_max_prompt_len_and_new_tokens(model: str,
3939
# shape
4040
prompt = create_text_prompt(model,
4141
min_token_length=max_prompt_length,
42-
max_token_length=max_prompt_length + max_new_tokens -
43-
1)
42+
max_token_length=max_prompt_length +
43+
max_new_tokens - 1)
4444
sampling_params = SamplingParams(max_tokens=1)
4545

4646
with pytest.raises(ValueError, match="warmup"):

0 commit comments

Comments
 (0)