Prevent nil pointer in pod helper (#6225)

Signed-off-by: Future-Outlier <[email protected]>
flyteorg · Feb 6, 2025 · b5443c8 · b5443c8
1 parent 66d7257
commit b5443c8
Showing 1 changed file with 2 additions and 2 deletions.
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 podTemplate.GetPodSpec() != nil {
+		if podTemplate != nil && podTemplate.GetPodSpec() != nil {
 			podSpec, objectMeta, err = ApplyPodTemplateOverride(objectMeta, podTemplate)
 			if err != nil {
 				return nil, nil, "", err
@@ -325,7 +325,7 @@ func BuildRawPod(ctx context.Context, tCtx pluginsCore.TaskExecutionContext) (*v
 
 		// handle pod template override
 		podTemplate := tCtx.TaskExecutionMetadata().GetOverrides().GetPodTemplate()
-		if podTemplate.GetPodSpec() != nil {
+		if podTemplate != nil && podTemplate.GetPodSpec() != nil {
 			podSpec, objectMeta, err = ApplyPodTemplateOverride(objectMeta, podTemplate)
 			if err != nil {
 				return nil, nil, "", err