From 0833f7a2cc16a512a97cbf7dc8c40e9232dd2d10 Mon Sep 17 00:00:00 2001 From: Philipp Moritz Date: Sun, 29 Aug 2021 23:18:37 -0700 Subject: [PATCH] Fix Ray client GOAWAY --- python/ray/util/client/worker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/ray/util/client/worker.py b/python/ray/util/client/worker.py index 113c7963e425..dcb7d54b2f19 100644 --- a/python/ray/util/client/worker.py +++ b/python/ray/util/client/worker.py @@ -504,7 +504,7 @@ def ping_server(self) -> bool: return False def is_connected(self) -> bool: - return self._conn_state == grpc.ChannelConnectivity.READY + return self._conn_state in [grpc.ChannelConnectivity.READY, grpc.ChannelConnectivity.IDLE] def _call_init(self, init_request: ray_client_pb2.InitRequest) -> None: init_resp = self.data_client.Init(init_request)