feat: add affinity template into agent (#126)

Signed-off-by: Anthony Rabbito <arabbito@coreweave.com> Co-authored-by: Ben Sherman <ben@wandb.com>
wandb · May 29, 2024 · 1ea5c6e · 1ea5c6e
1 parent b512d95
commit 1ea5c6e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 2 deletions.
diff --git a/charts/launch-agent/Chart.yaml b/charts/launch-agent/Chart.yaml
@@ -3,7 +3,7 @@ name: launch-agent
 icon: https://em-content.zobj.net/thumbs/240/apple/354/rocket_1f680.png
 description: A Helm chart for running the W&B Launch Agent in Kubernetes
 type: application
-version: 0.13.3
+version: 0.13.4
 maintainers:
   - name: wandb
     email: support@wandb.com

diff --git a/charts/launch-agent/templates/deployment.yaml b/charts/launch-agent/templates/deployment.yaml
@@ -148,6 +148,8 @@ spec:
         {{- toYaml .Values.agent.nodeSelector | nindent 8 }}
       tolerations:
         {{- toYaml .Values.agent.tolerations | nindent 8 }}
+      affinity:
+        {{- toYaml .Values.agent.affinity | nindent 8 }}
 ---
 {{- if .Capabilities.APIVersions.Has "policy/v1" }}
 apiVersion: policy/v1
@@ -160,4 +162,4 @@ spec:
     matchLabels:
       app: launch-agent-{{ .Release.Name }}
 ---
-{{- end }}
+{{- end }}
diff --git a/charts/launch-agent/values.yaml b/charts/launch-agent/values.yaml
@@ -21,6 +21,8 @@ agent:
   minAvailable: 1
   # Tolerations for the agent pod.
   tolerations: []
+  # Affinites for the agent pod.
+  affinity: {}
 
 # Namespace to deploy launch agent into
 namespace: wandb