예제 #1
0
파일: conftest.py 프로젝트: RuofanKong/ray
 def kill_raylet(ip, port, graceful=False):
     raylet_address = f"{ip}:{port}"
     channel = grpc.insecure_channel(raylet_address)
     stub = node_manager_pb2_grpc.NodeManagerServiceStub(channel)
     print(f"Sending a shutdown request to {ip}:{port}")
     stub.ShutdownRaylet(
         node_manager_pb2.ShutdownRayletRequest(graceful=graceful))
예제 #2
0
파일: test_utils.py 프로젝트: parasj/ray
 def _kill_raylet(self, ip, port, graceful=False):
     raylet_address = f"{ip}:{port}"
     channel = grpc.insecure_channel(raylet_address)
     stub = node_manager_pb2_grpc.NodeManagerServiceStub(channel)
     try:
         stub.ShutdownRaylet(
             node_manager_pb2.ShutdownRayletRequest(graceful=graceful))
     except _InactiveRpcError:
         assert not graceful
예제 #3
0
 def kill_raylet(ip, port, graceful=True):
     raylet_address = f"{ip}:{port}"
     channel = grpc.insecure_channel(raylet_address)
     stub = node_manager_pb2_grpc.NodeManagerServiceStub(channel)
     print(f"Sending a shutdown request to {ip}:{port}")
     try:
         stub.ShutdownRaylet(
             node_manager_pb2.ShutdownRayletRequest(graceful=graceful))
     except _InactiveRpcError:
         assert not graceful
예제 #4
0
 def _kill_raylet(self, ip, port, graceful=False):
     raylet_address = f"{ip}:{port}"
     channel = grpc.insecure_channel(raylet_address)
     stub = node_manager_pb2_grpc.NodeManagerServiceStub(channel)
     stub.ShutdownRaylet(
         node_manager_pb2.ShutdownRayletRequest(graceful=graceful))