[Core][BugFix] Fix docker runtime env on Azure (#3450)

* fix * format
skypilot-org · Apr 19, 2024 · cade827 · cade827
1 parent 20493fb
commit cade827
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/sky/clouds/azure.py b/sky/clouds/azure.py
@@ -294,7 +294,8 @@ def make_deploy_resources_variables(
         else:
             custom_resources = None
 
-        if resources.image_id is None:
+        if (resources.image_id is None or
+                resources.extract_docker_image() is not None):
             # pylint: disable=import-outside-toplevel
             from sky.clouds.service_catalog import azure_catalog
             gen_version = azure_catalog.get_gen_version_from_instance_type(

diff --git a/sky/clouds/gcp.py b/sky/clouds/gcp.py
@@ -461,8 +461,8 @@ def make_deploy_resources_variables(
                     # CUDA driver version 535.86.10, CUDA Library 12.2
                     image_id = 'skypilot:gpu-debian-11'
 
-        if resources.image_id is not None and resources.extract_docker_image(
-        ) is None:
+        if (resources.image_id is not None and
+                resources.extract_docker_image() is None):
             if None in resources.image_id:
                 image_id = resources.image_id[None]
             else: