Update kubeflow deployment to for standalone release

jupyter/terraform-gke/main.tf

@@ -25,8 +25,8 @@ module "gke" {
       machine_type              = "e2-medium"
       node_locations            = "us-central1-c"
       auto_scaling              = false
-      min_count                 = 1
-      max_count                 = 2
+      min_count                 = 0
+      max_count                 = 1
       local_ssd_count           = 0
       spot                      = false
       disk_size_gb              = 64
@@ -46,7 +46,7 @@ module "gke" {
       node_locations            = "us-central1-c"
       auto_scaling              = false
       min_count                 = 0
-      max_count                 = 4
+      max_count                 = 1
       local_ssd_count           = 0
       spot                      = false
       disk_size_gb              = 64

jupyter/terraform-kubeflow/main.tf

@@ -7,16 +7,27 @@ data "google_container_cluster" "testbed_cluster" {
 }
 
 # Setup kubeflow with kubeflow "kustomization" provided by WOGRA-AG
-module "kubeflow" {
-  source = "WOGRA-AG/kubeflow/kustomization"
+#module "kubeflow" {
+#  source = "WOGRA-AG/kubeflow/kustomization"
+#
+#  dex_user_email = "[email protected]"
+#
+#  # Disable 'production' features of Kubeflow. Change if need, e.g. "serving" for inference.
+#  deploy_notebooks = false
+#  deploy_dashboard = false
+#  deploy_katib = false
+#  deploy_tensorboard = false
+#  deploy_volumes = false
+#  deploy_serving = false
+#}
 
-  dex_user_email = "[email protected]"
 
-  # Disable 'production' features of Kubeflow. Change if need, e.g. "serving" for inference.
-  deploy_notebooks = false
-  deploy_dashboard = false
-  deploy_katib = false
-  deploy_tensorboard = false
-  deploy_volumes = false
-  deploy_serving = false
+data "kustomization_build" "training-operator" {
+  path = "github.com/kubeflow/manifests.git/apps/training-operator/upstream/overlays/standalone?ref=${var.kubeflow_version}"
 }
+
+resource "kustomization_resource" "training-operator" {
+  for_each = data.kustomization_build.training-operator.ids
+  manifest = data.kustomization_build.training-operator.manifests[each.value]
+
+}

jupyter/terraform-kubeflow/variables.tf

@@ -27,7 +27,19 @@ variable "description" {
 }
 
 variable "account_id" {
+  type = string
   description = "The service account Identifier to be used to interact with Google cloud."
   default = "terraform-iam-service-account"
 }
 
+variable "complete" {
+  type = bool
+  description = "Whether or not to fully install kubeflow, or only training operators."
+  default = false
+}
+
+variable "kubeflow_version" {
+  type = string
+  description = "Kubeflow (training operator) to install."
+  default = "v1.5.0"
+}