From b239a6cbf3588eeef5aa37c55cd9ae231eceb9d4 Mon Sep 17 00:00:00 2001
From: Christopher Cooper <cooperc@assemblesys.com>
Date: Thu, 31 Oct 2024 11:10:22 -0700
Subject: [PATCH] explicitly specify stages

---
 sky/execution.py | 12 +++++++-----
 1 file changed, 7 insertions(+), 5 deletions(-)

diff --git a/sky/execution.py b/sky/execution.py
index 69b5f439562..8fab5e583fb 100644
--- 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