Add nil check in databricks plugin (#332)

Signed-off-by: Kevin Su <pingsutw@apache.org>
flyteorg · May 8, 2023 · 76a80ec · 76a80ec
1 parent 1e0faab
commit 76a80ec
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/go/tasks/plugins/webapi/databricks/plugin.go b/go/tasks/plugins/webapi/databricks/plugin.go
@@ -168,6 +168,9 @@ func (p Plugin) Get(ctx context.Context, taskCtx webapi.GetContext) (latest weba
 	if err != nil {
 		return nil, err
 	}
+	if data == nil || data["state"] == nil {
+		return nil, errors.Errorf("CorruptedPluginState", "can't get the job state")
+	}
 	jobState := data["state"].(map[string]interface{})
 	message := fmt.Sprintf("%s", jobState["state_message"])
 	jobID := fmt.Sprintf("%.0f", data["job_id"])