From 8ca0a32819d1d45dc0d7b61a17baeecb4cbe153a Mon Sep 17 00:00:00 2001
From: Dmitri Gekhtman <62982571+DmitriGekhtman@users.noreply.github.com>
Date: Tue, 9 Feb 2021 22:34:24 -0800
Subject: [PATCH] HotFix k8s autoscaling (#14024)

---
 python/ray/autoscaler/_private/kubernetes/config.py | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/python/ray/autoscaler/_private/kubernetes/config.py b/python/ray/autoscaler/_private/kubernetes/config.py
index b285e7701..dcc315bc9 100644
--- a/python/ray/autoscaler/_private/kubernetes/config.py
+++ b/python/ray/autoscaler/_private/kubernetes/config.py
@@ -94,6 +94,11 @@ def get_autodetected_resources(container_data):
         for resource_name in ["cpu", "gpu"]
     }
 
+    # Throw out GPU from resource dict if the amount is 0.
+    for key in copy.deepcopy(node_type_resources):
+        if node_type_resources[key] == 0:
+            del node_type_resources[key]
+
     return node_type_resources