clean unused code.

Signed-off-by: zhanluxianshen <[email protected]>
volcengine · Feb 12, 2025 · 6a10a8d · 6a10a8d
1 parent dbbf95c
commit 6a10a8d
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/verl/single_controller/base/worker.py b/verl/single_controller/base/worker.py
@@ -42,10 +42,8 @@ def get_node_ip_by_sdk():
             if os.getenv("WG_BACKEND", None) == "ray":
                 import ray
                 return ray._private.services.get_node_ip_address()
-            elif os.getenv("WG_BACKEND", None) == "torch_rpc":
-                from verl.single_controller.torchrpc.k8s_client import get_ip_addr
-                return get_ip_addr()
-            return None
+            else:
+                raise NotImplementedError("WG_BACKEND now just support ray mode.")
 
         host_ipv4 = os.getenv("MY_HOST_IP", None)
         host_ipv6 = os.getenv("MY_HOST_IPV6", None)