helm-chart/ray-cluster: allow head autoscaling (ray-project#443)

Also allow setting rayVersion Signed-off-by: Christos Kotsis <[email protected]>
lowang-bh · Aug 9, 2022 · c802311 · c802311
1 parent 3033dd5
commit c802311
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/helm-chart/ray-cluster/templates/raycluster-cluster.yaml b/helm-chart/ray-cluster/templates/raycluster-cluster.yaml
@@ -5,6 +5,12 @@ metadata:
 {{ include "ray-cluster.labels" . | indent 4 }}
   name: {{ include "ray-cluster.fullname" . }}
 spec:
+  {{- if .Values.head.rayVersion }}
+  rayVersion: {{ .Values.head.rayVersion }}
+  {{- end }}
+  {{- if .Values.head.enableInTreeAutoscaling }}
+  enableInTreeAutoscaling: {{ .Values.head.enableInTreeAutoscaling }}
+  {{- end }}
   headGroupSpec:
     serviceType: ClusterIP
     rayStartParams: