reorder more imports for non-dragon build

CrayLabs · Sep 25, 2024 · 95c10b3 · 95c10b3
1 parent 5461da3
commit 95c10b3
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/dragon/test_worker_manager.py b/tests/dragon/test_worker_manager.py
@@ -31,8 +31,6 @@
 
 import pytest
 
-from smartsim._core.mli.comm.channel.dragon_util import create_local
-
 torch = pytest.importorskip("torch")
 dragon = pytest.importorskip("dragon")
 
@@ -56,6 +54,7 @@
 from dragon.data.ddict.ddict import DDict
 
 from smartsim._core.mli.comm.channel.dragon_fli import DragonFLIChannel
+from smartsim._core.mli.comm.channel.dragon_util import create_local
 from smartsim._core.mli.infrastructure.control.worker_manager import (
     EnvironmentConfigLoader,
     WorkerManager,