fix restarting

skypilot-org · Jul 5, 2024 · 16379f1 · 16379f1
1 parent 72bd0c6
commit 16379f1
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/sky/provision/azure/instance.py b/sky/provision/azure/instance.py
@@ -367,8 +367,7 @@ def _create_instance_tag(target_instance, is_head: bool = True) -> str:
                    _RESUME_PER_INSTANCE_TIMEOUT):
                 status = _get_instance_status(compute_client, inst,
                                               resource_group)
-                if status in (AzureInstanceStatus.RUNNING,
-                              AzureInstanceStatus.STOPPED):
+                if status == AzureInstanceStatus.STOPPED:
                     break
                 time.sleep(1)
             else:
@@ -378,6 +377,8 @@ def _create_instance_tag(target_instance, is_head: bool = True) -> str:
                     'Retrying ...')
                 stopping_instances.append(inst)
                 time.sleep(5)
+                continue
+            stopped_instances.append(inst)
         if stopping_instances and to_start_count > len(stopped_instances):
             msg = ('Timeout for waiting for existing instances '
                    f'{stopping_instances} in STOPPING state to '