fix: update topology spread constraints for custom worker pools (#2633)

Signed-off-by: TessaIO <[email protected]>
ray-project · Dec 11, 2024 · 9cd80cc · 9cd80cc
1 parent b5bcb86
commit 9cd80cc
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/helm-chart/ray-cluster/templates/raycluster-cluster.yaml b/helm-chart/ray-cluster/templates/raycluster-cluster.yaml
@@ -195,8 +195,8 @@ spec:
         {{ if $values.volumes }}
         volumes: {{- toYaml $values.volumes | nindent 10 }}
         {{- end }}
-        {{- if .Values.topologySpreadConstraints.enabled }}
-        topologySpreadConstraints: {{- toYaml .Values.worker.topologySpreadConstraints | nindent 10 }}
+        {{- if $values.topologySpreadConstraints }}
+        topologySpreadConstraints: {{- toYaml $values.topologySpreadConstraints | nindent 10 }}
         {{- end }}
         affinity: {{- toYaml $values.affinity | nindent 10 }}
         {{ if $values.priorityClassName }}

diff --git a/helm-chart/ray-cluster/values.yaml b/helm-chart/ray-cluster/values.yaml
@@ -268,6 +268,10 @@ additionalWorkerGroups:
     command: []
     args: []
 
+    # Topology Spread Constraints for worker pods
+    # See: https://kubernetes.io/docs/concepts/scheduling-eviction/topology-spread-constraints/
+    topologySpreadConstraints: {}
+
     # Custom pod DNS configuration
     # See https://kubernetes.io/docs/concepts/services-networking/dns-pod-service/#pod-dns-config
     # dnsConfig: