adjust import setuptools_rust (#10524)

2025-09-16 23:01:58 +08:00
parent 925dbb3218
commit 0abb41c70d
1 changed files with 2 additions and 1 deletions
--- a/sgl-router/setup.py
+++ b/sgl-router/setup.py
@@ -1,12 +1,13 @@
 import os

 from setuptools import setup
-from setuptools_rust import Binding, RustExtension

 no_rust = os.environ.get("SGLANG_ROUTER_BUILD_NO_RUST") == "1"

 rust_extensions = []
 if not no_rust:
+    from setuptools_rust import Binding, RustExtension
+
    rust_extensions.append(
        RustExtension(
            target="sglang_router_rs",