explicitly specify stages

skypilot-org · Oct 31, 2024 · b239a6c · b239a6c
1 parent 89fc09d
commit b239a6c
Showing 1 changed file with 7 additions and 5 deletions.
diff --git a/sky/execution.py b/sky/execution.py
@@ -473,11 +473,13 @@ def launch(
                     dryrun=dryrun)
                 handle = maybe_handle
                 # Get all stages
-                stages = list(Stage)
-                # Skip CLONE_DISK, PROVISION, and SETUP
-                stages.remove(Stage.CLONE_DISK)
-                stages.remove(Stage.PROVISION)
-                stages.remove(Stage.SETUP)
+                stages = [
+                    Stage.SYNC_WORKDIR,
+                    Stage.SYNC_FILE_MOUNTS,
+                    Stage.PRE_EXEC,
+                    Stage.EXEC,
+                    Stage.DOWN,
+                ]
             except exceptions.ClusterNotUpError:
                 # Proceed with normal provisioning
                 pass