Fix race condition connecting to docker ondemand_cluster

run-house · Jul 2, 2024 · 2ce4a89 · 2ce4a89
1 parent 221f3b6
commit 2ce4a89
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/runhouse/resources/hardware/sky_ssh_runner.py b/runhouse/resources/hardware/sky_ssh_runner.py
@@ -545,4 +545,8 @@ def ssh_tunnel(
     )
 
     cache_existing_sky_ssh_runner(address, ssh_port, runner)
+    if docker_user is not None:
+        # We needed this as of 2-Jul-24 when using Docker images on ondemand_clusters, otherwise
+        # we got a ConnectionError when trying to form the HTTP client.
+        time.sleep(0.5)
     return runner