Handle "stopping" runs similarly to "stopped" runs

trickest · Jul 26, 2024 · 18a2ca3 · 18a2ca3
1 parent a9e5400
commit 18a2ca3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/cmd/execute/watch.go b/cmd/execute/watch.go
@@ -115,7 +115,7 @@ func WatchRun(runID uuid.UUID, downloadPath string, nodesToDownload map[string]o
 			return
 		}
 
-		if run.Status == "COMPLETED" || run.Status == "STOPPED" || run.Status == "FAILED" {
+		if run.Status == "COMPLETED" || run.Status == "STOPPED" || run.Status == "STOPPING" || run.Status == "FAILED" {
 			if downloadPath == "" {
 				downloadPath = run.SpaceName
 				if run.ProjectName != "" {

diff --git a/cmd/output/output.go b/cmd/output/output.go
@@ -163,7 +163,7 @@ func init() {
 }
 
 func DownloadRunOutput(run *types.Run, nodes map[string]NodeInfo, files []string, destinationPath string) {
-	if run.Status != "COMPLETED" && run.Status != "STOPPED" && run.Status != "FAILED" {
+	if run.Status != "COMPLETED" && run.Status != "STOPPED" && run.Status != "STOPPING" && run.Status != "FAILED" {
 		fmt.Println("The workflow run hasn't been completed yet!")
 		fmt.Println("Run ID: " + run.ID.String() + "   Status: " + run.Status)
 		return