From 7fe2af41a306ebbf3c91491a7e1ecea84bd5afd8 Mon Sep 17 00:00:00 2001
From: Emma Ai <emma.ai@ga.gov.au>
Date: Mon, 25 Nov 2024 08:09:37 +0000
Subject: [PATCH] shutdown dask cluster gracefully

---
 odc/stats/proc.py | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/odc/stats/proc.py b/odc/stats/proc.py
index 4749846c..deceb182 100644
--- a/odc/stats/proc.py
+++ b/odc/stats/proc.py
@@ -327,6 +327,10 @@ def run(
             )
         raise ValueError("Must supply one of tasks= or sqs=")
 
+    def __del__(self):
+        if self.client():
+            self.client().close()
+
 
 def get_max_mem() -> int:
     """