Python ShutdownRayletRequestの例

コード例 #1

ファイル: 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

ファイル: 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

 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

 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))