diff --git a/sky/provision/gcp/mig_utils.py b/sky/provision/gcp/mig_utils.py index 9e33f5171e2..3a7d0db1805 100644 --- a/sky/provision/gcp/mig_utils.py +++ b/sky/provision/gcp/mig_utils.py @@ -207,3 +207,4 @@ def wait_for_managed_group_to_be_stable(project_id: str, zone: str, except subprocess.CalledProcessError as e: stderr = e.stderr.decode('ascii') logger.info(stderr) + raise diff --git a/sky/templates/gcp-ray.yml.j2 b/sky/templates/gcp-ray.yml.j2 index d986adbf6df..d7e787953d9 100644 --- a/sky/templates/gcp-ray.yml.j2 +++ b/sky/templates/gcp-ray.yml.j2 @@ -88,7 +88,9 @@ available_node_types: {%- if gcp_use_managed_instance_group %} managed-instance-group: run_duration: {{ run_duration }} + {%- if provision_timeout is defined and provision_timeout is not none %} provision_timeout: {{ provision_timeout }} + {%- endif %} {%- endif %} {%- if specific_reservations %} reservationAffinity: