Only override pod affinity

loft-sh · Sep 10, 2024 · f65d783 · f65d783
1 parent 65eca07
commit f65d783
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/pkg/kubernetes/run.go b/pkg/kubernetes/run.go
@@ -237,10 +237,11 @@ func (k *KubernetesDriver) runContainer(
 
 		// ensure we have a pod affinity, and in that case we have, just add ours
 		if pod.Spec.Affinity == nil || pod.Spec.Affinity.PodAffinity == nil {
-			pod.Spec.Affinity = &corev1.Affinity{
-				PodAffinity: &corev1.PodAffinity{
-					RequiredDuringSchedulingIgnoredDuringExecution: []corev1.PodAffinityTerm{},
-				},
+			if pod.Spec.Affinity == nil {
+				pod.Spec.Affinity = &corev1.Affinity{}
+			}
+			pod.Spec.Affinity.PodAffinity = &corev1.PodAffinity{
+				RequiredDuringSchedulingIgnoredDuringExecution: []corev1.PodAffinityTerm{},
 			}
 		}