Fix dp print message (#2138)

sgl-project · Nov 23, 2024 · 751c3a0 · 751c3a0
1 parent 60769be
commit 751c3a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/python/sglang/srt/managers/scheduler.py b/python/sglang/srt/managers/scheduler.py
@@ -1388,8 +1388,8 @@ def run_scheduler_process(
     pipe_writer,
 ):
     # [For Router] if env var "DP_RANK" exist, set dp_rank to the value of the env var
-    if dp_rank is None:
-        dp_rank = int(os.getenv("DP_RANK", -1))
+    if dp_rank is None and "DP_RANK" in os.environ:
+        dp_rank = int(os["DP_RANK"])
 
     if dp_rank is None:
         configure_logger(server_args, prefix=f" TP{tp_rank}")