Skip to content

Commit dd5a062

Browse files
committed
style: fix import line length in test file
Signed-off-by: Ye Yu <yey@nvidia.com> Signed-off-by: Ye Yu <yeyu@nvidia.com>
1 parent 922c2ea commit dd5a062

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

tests/unit/torch/speculative/plugins/test_hf_speculative_offline.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -34,7 +34,10 @@
3434

3535
import modelopt.torch.speculative as mtsp
3636
from modelopt.torch.speculative.eagle.default_config import default_eagle_config
37-
from modelopt.torch.speculative.eagle.utils import EagleOfflineDataCollator, OfflineSupervisedDataset
37+
from modelopt.torch.speculative.eagle.utils import (
38+
EagleOfflineDataCollator,
39+
OfflineSupervisedDataset,
40+
)
3841

3942
_mock_scripts = types.ModuleType("scripts")
4043
_mock_ar = types.ModuleType("scripts.ar_validate")

0 commit comments

Comments
 (0)