Skip to content

Commit 4846074

Browse files
committed
refine
Signed-off-by: Dylan Chen <[email protected]>
1 parent 470e192 commit 4846074

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/unittest/_torch/speculative/test_eagle3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
import os
33
import sys
44
import tempfile
5+
import unittest
56
from pathlib import Path
67
from unittest.mock import patch
78

@@ -374,7 +375,6 @@ def test_multi_eagle3(use_one_model: bool):
374375
pass
375376

376377

377-
<<<<<<< HEAD
378378
@pytest.mark.parametrize("disable_overlap_scheduler", [True, False])
379379
def test_eagle3_cuda_graph_padding(disable_overlap_scheduler: bool):
380380
"""Test CUDA graph padding with 3 requests and max_batch_size=4.

0 commit comments

Comments
 (0)