From 27b5bedc1525b7040aa96076bd3b04611becb013 Mon Sep 17 00:00:00 2001
From: Rui Vieira <ruidevieira@googlemail.com>
Date: Thu, 14 Nov 2024 21:53:28 +0000
Subject: [PATCH] feat(lmeval): Add overlay for Kueue support

---
 config/overlays/odh-kueue/kustomization.yaml | 13 +++++++++++++
 config/overlays/odh-kueue/params.env         | 11 +++++++++++
 config/overlays/odh-kueue/patch.yaml         | 14 ++++++++++++++
 3 files changed, 38 insertions(+)
 create mode 100644 config/overlays/odh-kueue/kustomization.yaml
 create mode 100644 config/overlays/odh-kueue/params.env
 create mode 100644 config/overlays/odh-kueue/patch.yaml

diff --git a/config/overlays/odh-kueue/kustomization.yaml b/config/overlays/odh-kueue/kustomization.yaml
new file mode 100644
index 00000000..28837c2c
--- /dev/null
+++ b/config/overlays/odh-kueue/kustomization.yaml
@@ -0,0 +1,13 @@
+---
+apiVersion: kustomize.config.k8s.io/v1beta1
+kind: Kustomization
+resources:
+  - ../../base
+
+patchesStrategicMerge:
+  - patch.yaml
+
+configMapGenerator:
+  - env: params.env
+    behavior: merge
+    name: config
diff --git a/config/overlays/odh-kueue/params.env b/config/overlays/odh-kueue/params.env
new file mode 100644
index 00000000..f7ff45d0
--- /dev/null
+++ b/config/overlays/odh-kueue/params.env
@@ -0,0 +1,11 @@
+trustyaiServiceImage=quay.io/trustyai/trustyai-service:latest
+trustyaiOperatorImage=quay.io/trustyai/trustyai-service-operator:latest
+oauthProxyImage=quay.io/openshift/origin-oauth-proxy:4.14.0
+kServeServerless=enabled
+lmes-driver-image=quay.io/trustyai/ta-lmes-driver:latest
+lmes-pod-image=quay.io/trustyai/ta-lmes-job:latest
+lmes-pod-checking-interval=10s
+lmes-image-pull-policy=Always
+lmes-max-batch-size=24
+lmes-default-batch-size=8
+lmes-detect-device=true
diff --git a/config/overlays/odh-kueue/patch.yaml b/config/overlays/odh-kueue/patch.yaml
new file mode 100644
index 00000000..6c7b1306
--- /dev/null
+++ b/config/overlays/odh-kueue/patch.yaml
@@ -0,0 +1,14 @@
+apiVersion: apps/v1
+kind: Deployment
+metadata:
+  name: controller-manager
+  namespace: system
+spec:
+  template:
+    spec:
+      containers:
+        - name: manager
+          args:
+            - --leader-elect
+            - --enable-services
+            - "TAS,LMES,JOB_MGR"