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