clean up imports

dask · Jun 15, 2022 · b09b1da · b09b1da
1 parent cacd7dc
commit b09b1da
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/distributed/tests/test_active_memory_manager.py b/distributed/tests/test_active_memory_manager.py
@@ -9,22 +9,20 @@
 
 import pytest
 
-from distributed import Event, Nanny, wait
+from distributed import Event, Nanny, Scheduler, Worker, wait
 from distributed.active_memory_manager import (
     ActiveMemoryManagerExtension,
     ActiveMemoryManagerPolicy,
     RetireWorker,
 )
 from distributed.core import Status
-from distributed.scheduler import Scheduler
 from distributed.utils_test import (
     assert_story,
     captured_logger,
     gen_cluster,
     inc,
     slowinc,
 )
-from distributed.worker import Worker
 
 NO_AMM_START = {"distributed.scheduler.active-memory-manager.start": False}