[Kubernetes] Fix kubernetes available GPUs (#3631)

Fix kubernetes available GPUs
skypilot-org · Jun 5, 2024 · 7a6df2f · 7a6df2f
1 parent 7692eaa
commit 7a6df2f
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/sky/clouds/service_catalog/kubernetes_catalog.py b/sky/clouds/service_catalog/kubernetes_catalog.py
@@ -136,15 +136,13 @@ def list_accelerators_realtime(
                     total_accelerators_capacity[
                         accelerator_name] += quantized_count
 
+            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)
-                if accelerator_name not in total_accelerators_available:
-                    total_accelerators_available[
-                        accelerator_name] = quantized_availability
-                else:
-                    total_accelerators_available[
-                        accelerator_name] += quantized_availability
+                total_accelerators_available[
+                    accelerator_name] += quantized_availability
 
     result = []