From 783c73b5d9f8408cbe3e5fb5d8ba0f530a237d6e Mon Sep 17 00:00:00 2001
From: empyreus <rjsouza1995@gmail.com>
Date: Tue, 5 May 2026 21:27:30 +0000
Subject: [PATCH] try to resolve single

---
 .azure-pipelines/templates/sglang-test.yml | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/.azure-pipelines/templates/sglang-test.yml b/.azure-pipelines/templates/sglang-test.yml
index 5a2a58f7..1ac1c581 100644
--- a/.azure-pipelines/templates/sglang-test.yml
+++ b/.azure-pipelines/templates/sglang-test.yml
@@ -92,7 +92,14 @@ steps:
     runRemoteArgs: '--container sglang-mscclpp-test'
     remoteScript: |
       export FLASHINFER_DISABLE_VERSION_CHECK=1
-      
+
+      # The fork's pyproject.toml excludes `benchmark*` from installed
+      # packages, but bench_serving.py imports from sglang.benchmark.datasets.
+      # Put the source tree on PYTHONPATH so the benchmark subpackage resolves.
+      if [ -d "$(pwd)/sglang/python/sglang/benchmark" ]; then
+        export PYTHONPATH="$(pwd)/sglang/python:${PYTHONPATH:-}"
+      fi
+
       echo "Directory: $(pwd)"
 
       # Launch sglang server in the background