Increase Nanny close timeout for test_spilling_local_cuda_cluster

rapidsai · Dec 1, 2023 · 53e3064 · 53e3064
1 parent 004185e
commit 53e3064
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dask_cuda/tests/test_proxy.py b/dask_cuda/tests/test_proxy.py
@@ -306,6 +306,7 @@ def task(x):
         n_workers=1,
         device_memory_limit="1B",
         jit_unspill=jit_unspill,
+        worker_class=IncreasedCloseTimeoutNanny,
         asynchronous=True,
     ) as cluster:
         async with Client(cluster, asynchronous=True) as client: