move msg_pump out of utils

CrayLabs · Oct 29, 2024 · 7ea00ab · 7ea00ab
1 parent 4edb55f
commit 7ea00ab
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 5 deletions.
diff --git a/...cture/control/dragon_utils/dragon_util.py → ...mli/infrastructure/control/dragon_util.py b/...cture/control/dragon_utils/dragon_util.py → ...mli/infrastructure/control/dragon_util.py
diff --git a/smartsim/_core/mli/infrastructure/control/dragon_utils/__init__.py b/smartsim/_core/mli/infrastructure/control/dragon_utils/__init__.py
diff --git a/...tructure/control/dragon_utils/msg_pump.py → tests/dragon_wlm/msg_pump.py b/...tructure/control/dragon_utils/msg_pump.py → tests/dragon_wlm/msg_pump.py
diff --git a/tests/dragon_wlm/test_request_dispatcher.py b/tests/dragon_wlm/test_request_dispatcher.py
@@ -53,12 +53,9 @@
 from smartsim._core.mli.comm.channel.dragon_channel import DragonCommChannel
 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.dragon_utils.dragon_util import (
+from smartsim._core.mli.infrastructure.control.dragon_util import (
     function_as_dragon_proc,
 )
-from smartsim._core.mli.infrastructure.control.dragon_utils.msg_pump import (
-    mock_messages,
-)
 from smartsim._core.mli.infrastructure.control.request_dispatcher import (
     RequestBatch,
     RequestDispatcher,
@@ -78,7 +75,8 @@
 from smartsim._core.mli.message_handler import MessageHandler
 from smartsim.log import get_logger
 
-from .utils.channel import FileSystemCommChannel
+from .channel import FileSystemCommChannel
+from .msg_pump import mock_messages
 
 logger = get_logger(__name__)