lint

Signed-off-by: Rafael Raposo <[email protected]>
flyteorg · Nov 15, 2024 · 6a371a0 · 6a371a0
1 parent 404056a
commit 6a371a0
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/flyteplugins/go/tasks/plugins/webapi/agent/plugin.go b/flyteplugins/go/tasks/plugins/webapi/agent/plugin.go
@@ -263,8 +263,8 @@ func (p *Plugin) Status(ctx context.Context, taskCtx webapi.StatusContext) (phas
 	resource := taskCtx.Resource().(ResourceWrapper)
 	taskInfo := &core.TaskInfo{Logs: resource.LogLinks, CustomInfo: resource.CustomInfo}
 	errorCode := pluginErrors.TaskFailedWithError
-	if resource.AgentError != nil && resource.AgentError.Code != "" {
-		errorCode = resource.AgentError.Code
+	if resource.AgentError != nil && resource.AgentError.GetCode() != "" {
+		errorCode = resource.AgentError.GetCode()
 	}
 
 	switch resource.Phase {

diff --git a/flyteplugins/go/tasks/plugins/webapi/agent/plugin_test.go b/flyteplugins/go/tasks/plugins/webapi/agent/plugin_test.go
@@ -276,8 +276,8 @@ func TestPlugin(t *testing.T) {
 		phase, err := plugin.Status(context.Background(), taskContext)
 		assert.NoError(t, err)
 		assert.Equal(t, pluginsCore.PhasePermanentFailure, phase.Phase())
-		assert.Equal(t, "ERROR: 500", phase.Err().Code)
-		assert.Equal(t, "failed to run the job: boom", phase.Err().Message)
+		assert.Equal(t, "ERROR: 500", phase.Err().GetCode())
+		assert.Equal(t, "failed to run the job: boom", phase.Err().GetMessage())
 	})
 
 	t.Run("test TaskExecution_FAILED Status Without Agent Error", func(t *testing.T) {
@@ -292,7 +292,7 @@ func TestPlugin(t *testing.T) {
 		phase, err := plugin.Status(context.Background(), taskContext)
 		assert.NoError(t, err)
 		assert.Equal(t, pluginsCore.PhasePermanentFailure, phase.Phase())
-		assert.Equal(t, pluginErrors.TaskFailedWithError, phase.Err().Code)
+		assert.Equal(t, pluginErrors.TaskFailedWithError, phase.Err().GetCode())
 	})
 
 	t.Run("test UNDEFINED Phase", func(t *testing.T) {