Support non-standard Spark container names (#2441)

Signed-off-by: jbhalodia-slack <[email protected]>
kubeflow · Feb 20, 2025 · 79264a4 · 79264a4
1 parent d10b8f5
commit 79264a4
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/internal/webhook/sparkpod_defaulter.go b/internal/webhook/sparkpod_defaulter.go
@@ -714,6 +714,7 @@ func findContainer(pod *corev1.Pod) int {
 		candidateContainerNames = append(candidateContainerNames, common.SparkExecutorContainerName, common.Spark3DefaultExecutorContainerName)
 	}
 
+	// return -1 if the pod is not a Driver or an Executor pod
 	if len(candidateContainerNames) == 0 {
 		return -1
 	}
@@ -725,7 +726,8 @@ func findContainer(pod *corev1.Pod) int {
 			}
 		}
 	}
-	return -1
+	// if no containers match the candidateContainerNames, assume the first container is the spark container.
+	return 0
 }
 
 func hasContainer(pod *corev1.Pod, container *corev1.Container) bool {