Migrating configure_kueue to helm

Author: agrawalkhushi18

.yamllint

@@ -8,6 +8,7 @@ yaml-files:
 ignore:
 - modules/management/kubectl-apply/manifests/*
 - examples/gke-a4x/nvidia-dra-driver.yaml
+- modules/management/kubectl-apply/raw-config-chart/templates/
 
 rules:
   # Defaults

modules/management/kubectl-apply/README.md

@@ -203,7 +203,7 @@ limitations under the License.
 
 | Name | Source | Version |
 |------|--------|---------|
-| <a name="module_configure_kueue"></a> [configure\_kueue](#module\_configure\_kueue) | ./kubectl | n/a |
+| <a name="module_configure_kueue"></a> [configure\_kueue](#module\_configure\_kueue) | ./helm_install | n/a |
 | <a name="module_install_gib"></a> [install\_gib](#module\_install\_gib) | ./kubectl | n/a |
 | <a name="module_install_gpu_operator"></a> [install\_gpu\_operator](#module\_install\_gpu\_operator) | ./helm_install | n/a |
 | <a name="module_install_jobset"></a> [install\_jobset](#module\_install\_jobset) | ./helm_install | n/a |

modules/management/kubectl-apply/main.tf

@@ -103,17 +103,32 @@ module "install_kueue" {
 }
 
 module "configure_kueue" {
-  source        = "./kubectl"
-  source_path   = local.install_kueue ? try(var.kueue.config_path, "") : null
-  template_vars = local.install_kueue ? try(var.kueue.config_template_vars, null) : null
-  depends_on    = [module.install_kueue]
+  source           = "./helm_install"
+  count            = local.install_kueue ? (try(var.kueue.config_path, "") != "" ? 1 : 0) : 0
+  release_name     = "kueue-config"
+  chart_name       = "${path.module}/raw-config-chart"
+  chart_version    = "0.1.0"
+  namespace        = "kueue-system"
+  create_namespace = true
+  wait             = false # Configuration resources (Queues) usually don't need wait
 
-  server_side_apply = true
-  wait_for_rollout  = true
+  values_yaml = [
+    yamlencode({
+      manifests = [
+        for doc in split("\n---", (
+          try(var.kueue.config_path, "") != ""
+          ? (length(try(var.kueue.config_template_vars, {})) > 0 || length(regexall("\\.tftpl$", var.kueue.config_path)) > 0
+            ? templatefile(var.kueue.config_path, try(var.kueue.config_template_vars, {}))
+          : file(var.kueue.config_path))
+          : ""
+        )) : trimspace(doc)
+        if length(trimspace(doc)) > 0
+      ]
+    })
+  ]
+
+  depends_on = [module.install_kueue]
 
-  providers = {
-    kubectl = kubectl
-  }
 }
 
 module "install_jobset" {

modules/management/kubectl-apply/raw-config-chart/Chart.yaml

@@ -0,0 +1,20 @@
+# Copyright 2026 "Google LLC"
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+apiVersion: v2
+name: raw-config-chart
+description: Generic chart to apply raw manifests
+type: application
+version: 0.1.0
+appVersion: "1.16.0"

modules/management/kubectl-apply/raw-config-chart/templates/manifests.yaml

@@ -0,0 +1,19 @@
+# Copyright 2026 "Google LLC"
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+{{- range $index, $manifest := .Values.manifests }}
+---
+# Manifest {{ $index }}
+{{ $manifest }}
+{{- end }}

modules/management/kubectl-apply/raw-config-chart/values.yaml

@@ -0,0 +1,15 @@
+# Copyright 2026 "Google LLC"
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#     http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+manifests: []