[helm] Independent setting of memory resources for migration pods (#3281

) * add separate resources for migration pod * default to global resources if not set
zenml-io · Jan 21, 2025 · dd91059 · dd91059
1 parent fec0956
commit dd91059
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/src/zenml/zen_server/deploy/helm/templates/server-db-job.yaml b/src/zenml/zen_server/deploy/helm/templates/server-db-job.yaml
@@ -104,10 +104,12 @@ spec:
           envFrom:
           - secretRef:
               name: {{ include "zenml.fullname" . }}-db-migration
-          {{- with .Values.resources }}
           resources:
-            {{- toYaml . | nindent 12 }}
-          {{- end }}
+            {{- if .Values.resourcesMigrationPod }}
+            {{- toYaml .Values.resourcesMigrationPod | nindent 12 }}
+            {{- else }}
+            {{- toYaml .Values.resources | nindent 12 }}
+            {{- end }}
       {{- with .Values.tolerations }}
       tolerations:
         {{- toYaml . | nindent 8 }}

diff --git a/src/zenml/zen_server/deploy/helm/values.yaml b/src/zenml/zen_server/deploy/helm/values.yaml
@@ -990,6 +990,10 @@ resources: {}
   #   cpu: 100m
   #   memory: 128Mi
 
+resourcesMigrationPod: {}
+  # Resource requests and limits for the database migration pod.
+  # If not set, falls back to the global resources configuration defined in .Values.resources
+
 autoscaling:
   enabled: false
   minReplicas: 1