Skip to content

Commit 5868935

Browse files
committed
fix tests
Signed-off-by: Izzy Putterman <[email protected]>
1 parent a1d1a6f commit 5868935

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
@@ -225,10 +225,7 @@ def test_deepseek_eagle3():
225225
pass
226226

227227

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

0 commit comments

Comments
 (0)