Use system Python for pipeline test (CuTeDSL in system site-packages)

2026-05-17 18:13:42 +00:00
parent 925e390b93
commit 955d7533f2
1 changed files with 2 additions and 2 deletions
--- a/tests/test_pipeline_real_weights.py
+++ b/tests/test_pipeline_real_weights.py
@@ -10,8 +10,8 @@ import glob

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

-# Direct import from vllm/ subdirectory (not the pip-installed vllm)
-from nvfp4_cutedsl import CuTeDSLMoERunner
+# Must be run with system Python (not venv) — CuTeDSL is in system site-packages
+from vllm.nvfp4_cutedsl import CuTeDSLMoERunner

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