From d4b5b58af5f8096740f58b56fccd0b20ee028343 Mon Sep 17 00:00:00 2001 From: Kevin Su Date: Thu, 11 Apr 2024 14:20:23 -0700 Subject: [PATCH] nit Signed-off-by: Kevin Su --- flyteplugins/go/tasks/plugins/webapi/databricks/plugin.go | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/flyteplugins/go/tasks/plugins/webapi/databricks/plugin.go b/flyteplugins/go/tasks/plugins/webapi/databricks/plugin.go index d3c12ed15c..55e08049f7 100644 --- 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":