Fix labels for GCP TPU

skypilot-org · Jun 11, 2024 · 3b8b040 · 3b8b040
1 parent 121ae2c
commit 3b8b040
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/sky/clouds/gcp.py b/sky/clouds/gcp.py
@@ -509,6 +509,9 @@ def make_deploy_resources_variables(
             ('gcp', 'managed_instance_group'), None)
         use_mig = managed_instance_group_config is not None
         resources_vars['gcp_use_managed_instance_group'] = use_mig
+        # Convert boolean to 0 or 1 in string, as GCP does not support boolean
+        # value in labels for TPU VM APIs.
+        resources_vars['gcp_use_managed_instance_group_value'] = str(int(use_mig))
         if use_mig:
             resources_vars.update(managed_instance_group_config)
         return resources_vars

diff --git a/sky/templates/gcp-ray.yml.j2 b/sky/templates/gcp-ray.yml.j2
@@ -80,7 +80,7 @@ available_node_types:
       {%- for label_key, label_value in labels.items() %}
         {{ label_key }}: {{ label_value|tojson }}
       {%- endfor %}
-        managed-instance-group: {{ gcp_use_managed_instance_group }}
+        managed-instance-group: {{ gcp_use_managed_instance_group_value|tojson }}
       {%- if gcp_use_managed_instance_group %}
       managed-instance-group:
         run_duration: {{ run_duration }}