Added timeout for closing dask cluster

seanlaw · seanlaw · commit b441a6c03fb4 · 2026-02-22T13:59:18.000-05:00
diff --git a/tests/test_aamp_ostinato.py b/tests/test_aamp_ostinato.py
@@ -17,7 +17,7 @@ def dask_cluster():
         worker_dashboard_address=None,
     )
     yield cluster.scheduler_address
-    cluster.close()
+    cluster.close(timeout=60)
 
 
 @pytest.mark.parametrize(
diff --git a/tests/test_aamp_stimp.py b/tests/test_aamp_stimp.py
@@ -23,7 +23,7 @@ def dask_cluster():
         worker_dashboard_address=None,
     )
     yield cluster.scheduler_address
-    cluster.close()
+    cluster.close(timeout=60)
 
 
 @pytest.mark.parametrize("T", T)
diff --git a/tests/test_aampdist.py b/tests/test_aampdist.py
@@ -16,7 +16,7 @@ def dask_cluster():
         worker_dashboard_address=None,
     )
     yield cluster.scheduler_address
-    cluster.close()
+    cluster.close(timeout=60)
 
 
 test_data = [
diff --git a/tests/test_aamped.py b/tests/test_aamped.py
@@ -18,7 +18,7 @@ def dask_cluster():
         worker_dashboard_address=None,
     )
     yield cluster.scheduler_address
-    cluster.close()
+    cluster.close(timeout=60)
 
 
 test_data = [
diff --git a/tests/test_maamped.py b/tests/test_maamped.py
@@ -18,7 +18,7 @@ def dask_cluster():
         worker_dashboard_address=None,
     )
     yield cluster.scheduler_address
-    cluster.close()
+    cluster.close(timeout=60)
 
 
 test_data = [
diff --git a/tests/test_mpdist.py b/tests/test_mpdist.py
@@ -18,7 +18,7 @@ def dask_cluster():
         worker_dashboard_address=None,
     )
     yield cluster.scheduler_address
-    cluster.close()
+    cluster.close(timeout=60)
 
 
 test_data = [
diff --git a/tests/test_mstumped.py b/tests/test_mstumped.py
@@ -20,7 +20,7 @@ def dask_cluster():
         worker_dashboard_address=None,
     )
     yield cluster.scheduler_address
-    cluster.close()
+    cluster.close(timeout=60)
 
 
 test_data = [
diff --git a/tests/test_non_normalized_decorator.py b/tests/test_non_normalized_decorator.py
@@ -70,7 +70,7 @@ def dask_cluster():
         worker_dashboard_address=None,
     )
     yield cluster.scheduler_address
-    cluster.close()
+    cluster.close(timeout=60)
 
 
 test_data = [
diff --git a/tests/test_ostinato.py b/tests/test_ostinato.py
@@ -19,7 +19,7 @@ def dask_cluster():
         worker_dashboard_address=None,
     )
     yield cluster.scheduler_address
-    cluster.close()
+    cluster.close(timeout=60)
 
 
 @pytest.mark.parametrize(
diff --git a/tests/test_stimp.py b/tests/test_stimp.py
@@ -23,7 +23,7 @@ def dask_cluster():
         worker_dashboard_address=None,
     )
     yield cluster.scheduler_address
-    cluster.close()
+    cluster.close(timeout=60)
 
 
 @pytest.mark.parametrize("T", T)
diff --git a/tests/test_stumped.py b/tests/test_stumped.py
@@ -21,7 +21,7 @@ def dask_cluster():
         worker_dashboard_address=None,
     )
     yield cluster.scheduler_address
-    cluster.close()
+    cluster.close(timeout=60)
 
 
 test_data = [

Original file line number	Diff line number	Diff line change
`@@ -17,7 +17,7 @@ def dask_cluster():`
`17`	`17`	`worker_dashboard_address=None,`
`18`	`18`	`)`
`19`	`19`	`yield cluster.scheduler_address`
`20`		`- cluster.close()`
	`20`	`+ cluster.close(timeout=60)`
`21`	`21`
`22`	`22`
`23`	`23`	`@pytest.mark.parametrize(`
Original file line number	Diff line number	Diff line change
`@@ -23,7 +23,7 @@ def dask_cluster():`
`23`	`23`	`worker_dashboard_address=None,`
`24`	`24`	`)`
`25`	`25`	`yield cluster.scheduler_address`
`26`		`- cluster.close()`
	`26`	`+ cluster.close(timeout=60)`
`27`	`27`
`28`	`28`
`29`	`29`	`@pytest.mark.parametrize("T", T)`
Original file line number	Diff line number	Diff line change
`@@ -16,7 +16,7 @@ def dask_cluster():`
`16`	`16`	`worker_dashboard_address=None,`
`17`	`17`	`)`
`18`	`18`	`yield cluster.scheduler_address`
`19`		`- cluster.close()`
	`19`	`+ cluster.close(timeout=60)`
`20`	`20`
`21`	`21`
`22`	`22`	`test_data = [`
Original file line number	Diff line number	Diff line change
`@@ -18,7 +18,7 @@ def dask_cluster():`
`18`	`18`	`worker_dashboard_address=None,`
`19`	`19`	`)`
`20`	`20`	`yield cluster.scheduler_address`
`21`		`- cluster.close()`
	`21`	`+ cluster.close(timeout=60)`
`22`	`22`
`23`	`23`
`24`	`24`	`test_data = [`
Original file line number	Diff line number	Diff line change
`@@ -20,7 +20,7 @@ def dask_cluster():`
`20`	`20`	`worker_dashboard_address=None,`
`21`	`21`	`)`
`22`	`22`	`yield cluster.scheduler_address`
`23`		`- cluster.close()`
	`23`	`+ cluster.close(timeout=60)`
`24`	`24`
`25`	`25`
`26`	`26`	`test_data = [`
Original file line number	Diff line number	Diff line change
`@@ -70,7 +70,7 @@ def dask_cluster():`
`70`	`70`	`worker_dashboard_address=None,`
`71`	`71`	`)`
`72`	`72`	`yield cluster.scheduler_address`
`73`		`- cluster.close()`
	`73`	`+ cluster.close(timeout=60)`
`74`	`74`
`75`	`75`
`76`	`76`	`test_data = [`
Original file line number	Diff line number	Diff line change
`@@ -19,7 +19,7 @@ def dask_cluster():`
`19`	`19`	`worker_dashboard_address=None,`
`20`	`20`	`)`
`21`	`21`	`yield cluster.scheduler_address`
`22`		`- cluster.close()`
	`22`	`+ cluster.close(timeout=60)`
`23`	`23`
`24`	`24`
`25`	`25`	`@pytest.mark.parametrize(`