shutdown dask cluster gracefully (#173)

emmaai · Emma Ai · web-flow · commit caa72ffcb381 · 2024-11-26T12:11:29.000+11:00
Co-authored-by: Emma Ai &lt;emma.ai@ga.gov.au&gt;
diff --git 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:
     """

Original file line number	Diff line number	Diff line change
`@@ -327,6 +327,10 @@ def run(`
`327`	`327`	`)`
`328`	`328`	`raise ValueError("Must supply one of tasks= or sqs=")`
`329`	`329`
	`330`	`+ def __del__(self):`
	`331`	`+ if self.client():`
	`332`	`+ self.client().close()`
	`333`	`+`
`330`	`334`
`331`	`335`	`def get_max_mem() -> int:`
`332`	`336`	`"""`