fix pod template override error

Signed-off-by: Nelson Chen <[email protected]>
flyteorg · Feb 6, 2025 · 66d7257 · 66d7257
1 parent 7dfe7d4
commit 66d7257
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/flyteplugins/go/tasks/pluginmachinery/flytek8s/pod_helper.go b/flyteplugins/go/tasks/pluginmachinery/flytek8s/pod_helper.go
@@ -289,7 +289,7 @@ func BuildRawPod(ctx context.Context, tCtx pluginsCore.TaskExecutionContext) (*v
 
 		// handle pod template override
 		podTemplate := tCtx.TaskExecutionMetadata().GetOverrides().GetPodTemplate()
-		if tCtx.TaskExecutionMetadata().GetOverrides().GetPodTemplate() != nil {
+		if podTemplate.GetPodSpec() != nil {
 			podSpec, objectMeta, err = ApplyPodTemplateOverride(objectMeta, podTemplate)
 			if err != nil {
 				return nil, nil, "", err