diff --git a/python/ray/util/client/worker.py b/python/ray/util/client/worker.py index 745b78b51..2c6c50fe4 100644 --- a/python/ray/util/client/worker.py +++ b/python/ray/util/client/worker.py @@ -65,6 +65,7 @@ class Worker: conn_attempts += 1 try: grpc.channel_ready_future(self.channel).result(timeout=timeout) + break except grpc.FutureTimeoutError: if conn_attempts >= connection_retries: raise ConnectionError("ray client connection timeout")