make helm dependencies configurable

Author: ighosh98

community/modules/management/dependencies-installer/README.md

@@ -50,6 +50,8 @@ limitations under the License.
 | <a name="input_cluster_id"></a> [cluster\_id](#input\_cluster\_id) | An identifier for the gke cluster resource with format projects/<project\_id>/locations/<region>/clusters/<name>. | `string` | n/a | yes |
 | <a name="input_gke_cluster_exists"></a> [gke\_cluster\_exists](#input\_gke\_cluster\_exists) | A static flag that signals to downstream modules that a cluster has been created. Needed by community/modules/scripts/kubernetes-operations. | `bool` | `false` | no |
 | <a name="input_gpu_operator"></a> [gpu\_operator](#input\_gpu\_operator) | Install [GPU Operator](https://docs.nvidia.com/datacenter/cloud-native/gpu-operator/latest/getting-started.html) which uses the [Kubernetes operator](https://kubernetes.io/docs/concepts/extend-kubernetes/operator/) to automate the management of all NVIDIA software components needed to provision GPU. | <pre>object({<br/>    install = optional(bool, false)<br/>    version = optional(string, "v25.3.0")<br/>  })</pre> | `{}` | no |
+| <a name="input_jobset"></a> [jobset](#input\_jobset) | Install [Jobset](https://github.com/kubernetes-sigs/jobset) which manages a group of K8s [jobs](https://kubernetes.io/docs/concepts/workloads/controllers/job/) as a unit. | <pre>object({<br/>    install = optional(bool, false)<br/>    version = optional(string, "v0.7.2")<br/>  })</pre> | `{}` | no |
+| <a name="input_kueue"></a> [kueue](#input\_kueue) | Install and configure [Kueue](https://kueue.sigs.k8s.io/docs/overview/) workload scheduler. A configuration yaml/template file can be provided with config\_path to be applied right after kueue installation. If a template file provided, its variables can be set to config\_template\_vars. | <pre>object({<br/>    install              = optional(bool, false)<br/>    version              = optional(string, "v0.11.4")<br/>    config_path          = optional(string, null)<br/>    config_template_vars = optional(map(any), null)<br/>  })</pre> | `{}` | no |
 | <a name="input_nvidia_dra_driver"></a> [nvidia\_dra\_driver](#input\_nvidia\_dra\_driver) | Installs [Nvidia DRA driver](https://github.com/NVIDIA/k8s-dra-driver-gpu) which supports Dynamic Resource Allocation for NVIDIA GPUs in Kubernetes | <pre>object({<br/>    install = optional(bool, false)<br/>    version = optional(string, "v25.3.0-rc.2")<br/>  })</pre> | `{}` | no |
 | <a name="input_project_id"></a> [project\_id](#input\_project\_id) | The project ID that hosts the gke cluster. | `string` | n/a | yes |
 

community/modules/management/dependencies-installer/main.tf

@@ -39,7 +39,7 @@ module "install_kueue" {
   release_name = "kueue"
 
   chart_name    = "oci://registry.k8s.io/kueue/charts/kueue"
-  chart_version = "0.11.4" # Specify your desired Kueue version
+  chart_version = var.kueue.version # Specify your desired Kueue version
 
   create_namespace = true # Helm can also create the namespace
   wait             = true
@@ -48,10 +48,10 @@ module "install_kueue" {
 
 module "install_jobset" {
   source           = "./helm_install"
-  depends_on       = [var.gke_cluster_exists]
+  depends_on       = [var.gke_cluster_exists, module.install_kueue]
   release_name     = "jobset-controller"                          # The release name for your JobSet installation
   chart_name       = "oci://registry.k8s.io/jobset/charts/jobset" # The Helm repository URL for nvidia charts
-  chart_version    = "0.5.2"
+  chart_version    = var.jobset.version
   create_namespace = true
   namespace        = "jobset-system"
 }

community/modules/management/dependencies-installer/variables.tf

@@ -48,3 +48,23 @@ variable "nvidia_dra_driver" {
   })
   default = {}
 }
+
+variable "kueue" {
+  description = "Install and configure [Kueue](https://kueue.sigs.k8s.io/docs/overview/) workload scheduler. A configuration yaml/template file can be provided with config_path to be applied right after kueue installation. If a template file provided, its variables can be set to config_template_vars."
+  type = object({
+    install              = optional(bool, false)
+    version              = optional(string, "v0.11.4")
+    config_path          = optional(string, null)
+    config_template_vars = optional(map(any), null)
+  })
+  default = {}
+}
+
+variable "jobset" {
+  description = "Install [Jobset](https://github.com/kubernetes-sigs/jobset) which manages a group of K8s [jobs](https://kubernetes.io/docs/concepts/workloads/controllers/job/) as a unit."
+  type = object({
+    install = optional(bool, false)
+    version = optional(string, "v0.7.2")
+  })
+  default = {}
+}