From e932498f494d64adf86813e74cb142b1c139f94e Mon Sep 17 00:00:00 2001 From: Zhanghao Wu Date: Mon, 22 Apr 2024 19:36:11 +0000 Subject: [PATCH] format --- sky/backends/cloud_vm_ray_backend.py | 9 +++++---- 1 file changed, 5 insertions(+), 4 deletions(-) diff --git a/sky/backends/cloud_vm_ray_backend.py b/sky/backends/cloud_vm_ray_backend.py index 834eb397f67..d6e7528135b 100644 --- a/sky/backends/cloud_vm_ray_backend.py +++ b/sky/backends/cloud_vm_ray_backend.py @@ -269,10 +269,11 @@ def get_or_fail(futures, pg) -> List[int]: returncodes = [1] * len(futures) # Wait for 1 task to be ready. ready = [] - # Recall ray.wait if ready is empty. This is because ray.wait - # with timeout=None will only wait for 10**6 seconds, which will - # cause the task longer than 12 days returned before it is - # ready. Reference: https://github.com/ray-project/ray/blob/ray-2.9.3/python/ray/_private/worker.py#L2845-L2846 + # Keep invoking ray.wait if ready is empty. This is because + # ray.wait with timeout=None will only wait for 10**6 seconds, + # which will cause the task longer than 12 days returned before + # it is ready. + # Reference: https://github.com/ray-project/ray/blob/ray-2.9.3/python/ray/_private/worker.py#L2845-L2846 while not ready: ready, unready = ray.wait(futures) idx = futures.index(ready[0])