fix

skypilot-org · Aug 26, 2024 · c79abce · c79abce
1 parent 6067e2b
commit c79abce
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/sky/provision/runpod/utils.py b/sky/provision/runpod/utils.py
@@ -77,7 +77,11 @@ def list_instances() -> Dict[str, Dict[str, Any]]:
         info['name'] = instance['name']
         info['port2endpoint'] = {}
 
-        if instance['desiredStatus'] == 'RUNNING' and instance.get('runtime'):
+        # Sometimes when the cluster is in the process of being created,
+        # the `port` field in the runtime is None and we need to check for it.
+        if (instance['desiredStatus'] == 'RUNNING' and
+                instance.get('runtime') and
+                instance.get('runtime').get('ports')):
             for port in instance['runtime']['ports']:
                 if port['isIpPublic']:
                     if port['privatePort'] == 22: