Import cleanup fixture in test_dask_cuda_worker.py

rapidsai · May 31, 2022 · daf32be · daf32be
1 parent 99add03
commit daf32be
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/dask_cuda/tests/test_dask_cuda_worker.py b/dask_cuda/tests/test_dask_cuda_worker.py
@@ -10,8 +10,7 @@
 
 from distributed import Client, wait
 from distributed.system import MEMORY_LIMIT
-from distributed.utils_test import loop  # noqa: F401
-from distributed.utils_test import popen
+from distributed.utils_test import cleanup, loop, popen  # noqa: F401
 
 from dask_cuda.utils import (
     get_gpu_count_mig,