diff --git a/dask_cuda/tests/test_proxy.py b/dask_cuda/tests/test_proxy.py index 7614219bf..5458c5bab 100644 --- 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: