Fix import: use direct import from vllm/ subdirectory

2026-05-17 18:12:53 +00:00
parent cd6144b832
commit 925e390b93
1 changed files with 2 additions and 2 deletions
--- a/tests/test_pipeline_real_weights.py
+++ b/tests/test_pipeline_real_weights.py
@@ -9,9 +9,9 @@ import os
 import glob

 sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)) + '/..')
-sys.path.insert(0, os.path.dirname(os.path.abspath(__file__)) + '/../vllm')

-from vllm.nvfp4_cutedsl import CuTeDSLMoERunner
+# Direct import from vllm/ subdirectory (not the pip-installed vllm)
+from nvfp4_cutedsl import CuTeDSLMoERunner

 # ============================================================
 # CONFIG