helm-chart/ray-cluster: Allow setting pod lifecycle (ray-project#494)

lowang-bh · Aug 20, 2022 · 2ed0f11 · 2ed0f11
1 parent b792bb6
commit 2ed0f11
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/helm-chart/ray-cluster/templates/raycluster-cluster.yaml b/helm-chart/ray-cluster/templates/raycluster-cluster.yaml
@@ -37,6 +37,10 @@ spec:
             {{- if .Values.head.ports }}
             ports: {{- toYaml .Values.head.ports | nindent 14}}
             {{- end }}
+            {{- if .Values.head.lifecycle }}
+            lifecycle:
+            {{- toYaml .Values.head.lifecycle | nindent 14 }}
+            {{- end }}
         volumes: {{- toYaml .Values.head.volumes | nindent 10 }}
         affinity: {{- toYaml .Values.head.affinity | nindent 10 }}
         tolerations: {{- toYaml .Values.head.tolerations | nindent 10 }}
@@ -81,6 +85,10 @@ spec:
             envFrom: {{- toYaml $ | nindent 14}}
             {{- end }}
             ports: {{- toYaml $values.ports | nindent 14}}
+            {{- if $values.lifecycle }}
+            lifecycle:
+            {{- toYaml $values.lifecycle | nindent 14 }}
+            {{- end }}
         volumes: {{- toYaml $values.volumes | nindent 10 }}
         affinity: {{- toYaml $values.affinity | nindent 10 }}
         tolerations: {{- toYaml $values.tolerations | nindent 10 }}
@@ -124,6 +132,10 @@ spec:
             envFrom: {{- toYaml . | nindent 14}}
             {{- end }}
             ports: {{- toYaml .Values.worker.ports | nindent 14}}
+            {{- if .Values.worker.lifecycle }}
+            lifecycle:
+            {{- toYaml .Values.worker.lifecycle | nindent 14 }}
+            {{- end }}
         volumes: {{- toYaml .Values.worker.volumes | nindent 10 }}
         affinity: {{- toYaml .Values.worker.affinity | nindent 10 }}
         tolerations: {{- toYaml .Values.worker.tolerations | nindent 10 }}