Skip to content

Commit 15fad81

Browse files
committed
fix tests
Signed-off-by: Izzy Putterman <[email protected]>
1 parent 5c41752 commit 15fad81

File tree

1 file changed

+1
-4
lines changed

1 file changed

+1
-4
lines changed

tests/unittest/_torch/speculative/test_eagle3.py

Lines changed: 1 addition & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -226,10 +226,7 @@ def test_deepseek_eagle3():
226226
pass
227227

228228

229-
@pytest.mark.parametrize("use_one_model", [
230-
[True],
231-
[False],
232-
])
229+
@pytest.mark.parametrize("use_one_model", [True, False])
233230
def test_multi_eagle3(use_one_model: bool):
234231
use_cuda_graph = True
235232
attn_backend = "TRTLLM"

0 commit comments

Comments
 (0)