diff --git a/tests/test_pipeline_real_weights.py b/tests/test_pipeline_real_weights.py index 1a5a8144..61de57a0 100644 --- 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