From 925e390b93825dfc6b3b45555b93209bf61bf368 Mon Sep 17 00:00:00 2001
From: biondizzle <biondizzle@gmail.com>
Date: Sun, 17 May 2026 18:12:53 +0000
Subject: [PATCH] Fix import: use direct import from vllm/ subdirectory

---
 tests/test_pipeline_real_weights.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

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