From 587deb15a7ea1764c09dfd50196112ac8cf3e98a Mon Sep 17 00:00:00 2001
From: Baizhou Zhang <sobereddiezhang@gmail.com>
Date: Tue, 28 Oct 2025 20:07:36 -0700
Subject: [PATCH] [hotfix] Fix pytest not found in CI (#12311)

---
 python/pyproject.toml               | 1 +
 scripts/ci/ci_install_dependency.sh | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/python/pyproject.toml b/python/pyproject.toml
index 9701ed834..9afaeba37 100755
--- a/python/pyproject.toml
+++ b/python/pyproject.toml
@@ -89,6 +89,7 @@ test = [
   "sentence_transformers",
   "tabulate",
 ]
+dev = ["sglang[test]"]
 tracing = [
   "opentelemetry-api",
   "opentelemetry-exporter-otlp",
diff --git a/scripts/ci/ci_install_dependency.sh b/scripts/ci/ci_install_dependency.sh
index f800e0cb4..3130aceb0 100755
--- a/scripts/ci/ci_install_dependency.sh
+++ b/scripts/ci/ci_install_dependency.sh
@@ -55,7 +55,7 @@ else
     $PIP_CMD install flashinfer-python==0.4.1 --prerelease=allow $PIP_INSTALL_SUFFIX
 
     # Install the main package
-    $PIP_CMD install -e "python" --extra-index-url https://download.pytorch.org/whl/${CU_VERSION} $PIP_INSTALL_SUFFIX --upgrade
+    $PIP_CMD install -e "python[dev]" --extra-index-url https://download.pytorch.org/whl/${CU_VERSION} $PIP_INSTALL_SUFFIX --upgrade
 fi
 
 # Install router for pd-disagg test