fix missing fixture import; see pydata/xarray#6650

keflavich · Jun 12, 2022 · 30de5ea · 30de5ea
1 parent 3f69c11
commit 30de5ea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/spectral_cube/tests/test_dask.py b/spectral_cube/tests/test_dask.py
@@ -12,7 +12,7 @@
 from astropy.utils import data
 
 try:
-    from distributed.utils_test import client, loop, cluster_fixture  # noqa
+    from distributed.utils_test import client, loop, cluster_fixture, cleanup  # noqa
     DISTRIBUTED_INSTALLED = True
 except ImportError:
     DISTRIBUTED_INSTALLED = False