From 8f48c64507319585c249e48f0c4ddb7d614986ee Mon Sep 17 00:00:00 2001 From: Barak Michener Date: Wed, 13 Jan 2021 13:36:01 -0800 Subject: [PATCH] [ray_client]: Fix multiple attempts at checking connection (#13422) --- python/ray/util/client/worker.py | 1 + 1 file changed, 1 insertion(+) 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")