F OpenNebula/one-aiops#569: add timeout for waiting nodes

OpenNebula · Dec 18, 2024 · 55c5eae · 55c5eae
1 parent c70dacb
commit 55c5eae
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/lithops/serverless/backends/one/one.py b/lithops/serverless/backends/one/one.py
@@ -188,8 +188,9 @@ def deploy_runtime(self, docker_image_name, memory, timeout):
             self._scale_oneke(self.nodes, 1)
         return super().deploy_runtime(docker_image_name, memory, timeout)
 
-    def _wait_kubernetes_nodes(self, total):
+    def _wait_kubernetes_nodes(self, total, timeout=60):
         super()._get_nodes()
-        while len(self.nodes) < total:
+        start_time = time.time()
+        while len(self.nodes) < total and time.time() - start_time < timeout:
             time.sleep(1)
             super()._get_nodes()