fix: update topology spread constraints for custom worker pools

Signed-off-by: TessaIO <[email protected]>
ray-project · Dec 10, 2024 · dc569bb · dc569bb
1 parent 86506d6
commit dc569bb
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 4 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
@@ -133,7 +133,7 @@ head:
 worker:
   # If you want to disable the default workergroup
   # uncomment the line below
-  # disabled: true
+  disabled: true
   groupName: workergroup
   replicas: 1
   minReplicas: 1
@@ -212,7 +212,7 @@ worker:
 additionalWorkerGroups:
   smallGroup:
     # Disabled by default
-    disabled: true
+    disabled: false
     replicas: 0
     minReplicas: 0
     maxReplicas: 3
@@ -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: