nit

Signed-off-by: Kevin Su <[email protected]>
flyteorg · Apr 11, 2024 · d4b5b58 · d4b5b58
1 parent 0dcfb3b
commit d4b5b58
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/flyteplugins/go/tasks/plugins/webapi/databricks/plugin.go b/flyteplugins/go/tasks/plugins/webapi/databricks/plugin.go
@@ -263,10 +263,16 @@ func (p Plugin) Status(ctx context.Context, taskCtx webapi.StatusContext) (phase
 	taskInfo := createTaskInfo(exec.RunID, jobID, exec.DatabricksInstance)
 	switch lifeCycleState {
 	// Job response format. https://docs.databricks.com/en/workflows/jobs/jobs-2.0-api.html#runlifecyclestate
+	case "QUEUED":
+		return core.PhaseInfoQueued(time.Now(), core.DefaultPhaseVersion, message), nil
 	case "PENDING":
 		return core.PhaseInfoInitializing(time.Now(), core.DefaultPhaseVersion, message, taskInfo), nil
 	case "RUNNING":
 		fallthrough
+	case "BLOCKED":
+		fallthrough
+	case "WAITING_FOR_RETRY":
+		fallthrough
 	case "TERMINATING":
 		return core.PhaseInfoRunning(core.DefaultPhaseVersion, taskInfo), nil
 	case "TERMINATED":