untested revive_node() Python call

rodrigo-castellon · Jun 3, 2024 · 392aa33 · 392aa33
1 parent eaa9266
commit 392aa33
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/python/babyray/__init__.py b/python/babyray/__init__.py
@@ -151,6 +151,26 @@ def kill_node(node_id):
     node_worker_gRPC.KillServer(rayclient_pb2.StatusResponse())
 
 
+def revive_node(node_id):
+    # revive node node_id
+
+    local_scheduler_channel = grpc.insecure_channel(
+        f"node{node_id}:{str(LOCAL_SCHEDULER_PORT)}"
+    )
+    node_local_scheduler_gRPC = rayclient_pb2_grpc.LocalSchedulerStub(
+        local_scheduler_channel
+    )
+
+    worker_channel = grpc.insecure_channel(
+        f"node{node_id}:{str(LOCAL_WORKER_PORT_START)}"
+    )
+
+    node_worker_gRPC = rayclient_pb2_grpc.WorkerStub(worker_channel)
+
+    node_local_scheduler_gRPC.ReviveServer(rayclient_pb2.StatusResponse())
+    node_worker_gRPC.ReviveServer(rayclient_pb2.StatusResponse())
+
+
 def demo():
     # Example function to simulate Ray's behavior
     @remote