Minor update switched None to nan for cpu_count

skypilot-org · Jan 2, 2024 · 6cd932f · 6cd932f
1 parent 19edb99
commit 6cd932f
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/sky/clouds/service_catalog/common.py b/sky/clouds/service_catalog/common.py
@@ -530,8 +530,9 @@ def make_list_from_df(rows):
             ),
             axis='columns',
         ).tolist()
-        ret.sort(key=lambda info: (info.accelerator_count, info.price, info.
-                                   spot_price, info.region))
+        ret.sort(key=lambda info: (info.accelerator_count, info.cpu_count
+                                   if not pd.isna(info.cpu_count) else 0, info.
+                                   price, info.spot_price, info.region))
         return ret
 
     return {k: make_list_from_df(v) for k, v in grouped}