From 1ea5c6eeeefc07cf08fc3fcea2d7615c8bc2560e Mon Sep 17 00:00:00 2001 From: Anthony Rabbito Date: Wed, 29 May 2024 14:08:15 -0400 Subject: [PATCH] feat: add affinity template into agent (#126) Signed-off-by: Anthony Rabbito Co-authored-by: Ben Sherman --- charts/launch-agent/Chart.yaml | 2 +- charts/launch-agent/templates/deployment.yaml | 4 +++- charts/launch-agent/values.yaml | 2 ++ 3 files changed, 6 insertions(+), 2 deletions(-) diff --git a/charts/launch-agent/Chart.yaml b/charts/launch-agent/Chart.yaml index b8b328aa..1049e168 100644 --- 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 index 912e9db1..15d1e754 100644 --- 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 }} \ No newline at end of file +{{- end }} diff --git a/charts/launch-agent/values.yaml b/charts/launch-agent/values.yaml index b39f309d..529a6789 100644 --- 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