accept both

skypilot-org · Sep 18, 2023 · 605943c · 605943c
1 parent e5e400b
commit 605943c
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/sky/skylet/providers/command_runner.py b/sky/skylet/providers/command_runner.py
@@ -143,7 +143,9 @@ def run_init(self, *, as_head: bool, file_mounts: Dict[str, str],
                 docker_login_config.password,
                 docker_login_config.server,
             ))
-            specific_image = f'{docker_login_config.server}/{specific_image}'
+            server_prefix = f'{docker_login_config.server}/'
+            if not specific_image.startswith(server_prefix):
+                specific_image = f'{server_prefix}{specific_image}'
 
         if self.docker_config.get('pull_before_run', True):
             assert specific_image, ('Image must be included in config if ' +