[k8s] Fix show-gpus availability map when nvidia drivers are not in…

…stalled (#4429) * Fix availability map * Fix availability map
skypilot-org · Dec 3, 2024 · 747382a · 747382a
1 parent 5c1cb9c
commit 747382a
Showing 1 changed file with 3 additions and 4 deletions.
diff --git a/sky/clouds/service_catalog/kubernetes_catalog.py b/sky/clouds/service_catalog/kubernetes_catalog.py
@@ -239,13 +239,12 @@ def _list_accelerators(
 
                 accelerators_available = accelerator_count - allocated_qty
 
-                if accelerator_name not in total_accelerators_available:
-                    total_accelerators_available[accelerator_name] = 0
                 if accelerators_available >= min_quantity_filter:
                     quantized_availability = min_quantity_filter * (
                         accelerators_available // min_quantity_filter)
-                    total_accelerators_available[
-                        accelerator_name] += quantized_availability
+                    total_accelerators_available[accelerator_name] = (
+                        total_accelerators_available.get(accelerator_name, 0) +
+                        quantized_availability)
 
     result = []