Set a default value on idleTimeout (#884)

dask · Apr 19, 2024 · 7326d3c · 7326d3c
1 parent 67a45e9
commit 7326d3c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/dask_kubernetes/operator/controller/controller.py b/dask_kubernetes/operator/controller/controller.py
@@ -851,7 +851,8 @@ async def daskautoscaler_adapt(spec, name, namespace, logger, **kwargs):
 
 @kopf.timer("daskcluster.kubernetes.dask.org", interval=5.0)
 async def daskcluster_autoshutdown(spec, name, namespace, logger, **kwargs):
-    if spec["idleTimeout"]:
+    idle_timeout = spec.get("idleTimeout", 0)
+    if idle_timeout:
         try:
             idle_since = await check_scheduler_idle(
                 scheduler_service_name=f"{name}-scheduler",
@@ -861,6 +862,6 @@ async def daskcluster_autoshutdown(spec, name, namespace, logger, **kwargs):
         except Exception:
             logger.warn("Unable to connect to scheduler, skipping autoshutdown check.")
             return
-        if idle_since and time.time() > idle_since + spec["idleTimeout"]:
+        if idle_since and time.time() > idle_since + idle_timeout:
             cluster = await DaskCluster.get(name, namespace=namespace)
             await cluster.delete()