From 59f70ad351b9c31334549f479b579d9d68ac4be1 Mon Sep 17 00:00:00 2001
From: Frank Dekervel <kervel@gmail.com>
Date: Mon, 17 Mar 2025 13:21:02 +0100
Subject: [PATCH] allow setting kubernetes runtimeClass in the helm chart

---
 charts/clearml-agent/templates/agentk8sglue-configmap.yaml | 3 +++
 charts/clearml-agent/values.yaml                           | 3 +++
 2 files changed, 6 insertions(+)

diff --git a/charts/clearml-agent/templates/agentk8sglue-configmap.yaml b/charts/clearml-agent/templates/agentk8sglue-configmap.yaml
index e33656a..d768450 100644
--- a/charts/clearml-agent/templates/agentk8sglue-configmap.yaml
+++ b/charts/clearml-agent/templates/agentk8sglue-configmap.yaml
@@ -30,6 +30,9 @@ data:
       priorityClassName: {{ .Values.agentk8sglue.basePodTemplate.priorityClassName }}
       initContainers:
         {{- toYaml .Values.agentk8sglue.basePodTemplate.initContainers | nindent 8 }}
+      {{- if .values.agentk8sglue.basePodTemplate.runtimeClassName }}
+      runtimeClassName: {{ .values.agentk8sglue.basePodTemplate.runtimeClassName }}
+      {{- end }}
       containers:
       - resources:
           {{- toYaml .Values.agentk8sglue.basePodTemplate.resources | nindent 10 }}
diff --git a/charts/clearml-agent/values.yaml b/charts/clearml-agent/values.yaml
index eeeaf17..981d0b0 100644
--- a/charts/clearml-agent/values.yaml
+++ b/charts/clearml-agent/values.yaml
@@ -57,6 +57,9 @@ agentk8sglue:
   # -- Glue Agent number of pods
   replicaCount: 1
 
+  # -- kubernetes runtime class name
+  runtimeClassName: ""
+
   # -- Glue Agent pod resources
   resources: {}