fix skypilot output flavor names (#114)

zenml-io · Nov 28, 2023 · 3aa7e43 · 3aa7e43
1 parent c881e8c
commit 3aa7e43
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/mlstacks/terraform/aws-modular/output_file.tf b/src/mlstacks/terraform/aws-modular/output_file.tf
@@ -29,7 +29,7 @@ resource "local_file" "stack_file" {
 %{endif}
 
       orchestrator:
-%{if var.enable_orchestrator_kubeflow}      
+%{if var.enable_orchestrator_kubeflow}
         id: ${uuid()}
         flavor: kubeflow
         name: eks_kubeflow_orchestrator
@@ -43,7 +43,7 @@ resource "local_file" "stack_file" {
 %{else}
 %{if var.enable_orchestrator_skypilot}
         id: ${uuid()}
-        flavor: vm-aws
+        flavor: vm_aws
         name: aws_skypilot_orchestrator
         configuration: {}
 %{else}

diff --git a/src/mlstacks/terraform/gcp-modular/output_file.tf b/src/mlstacks/terraform/gcp-modular/output_file.tf
@@ -49,7 +49,7 @@ resource "local_file" "stack_file" {
 %{else}
 %{if var.enable_orchestrator_skypilot}
         id: ${uuid()}
-        flavor: vm-gcp
+        flavor: vm_gcp
         name: gcp_skypilot_orchestrator
         configuration: {"project_id": "${var.project_id}"}
 %{else}