WIP: [no ci]

nv-morpheus · Oct 5, 2023 · 64cd7f8 · 64cd7f8
1 parent 5d06a45
commit 64cd7f8
Show file tree

Hide file tree

Showing 4 changed files with 3 additions and 3 deletions.
diff --git a/morpheus/pipeline/__init__.py b/morpheus/pipeline/__init__.py
@@ -22,7 +22,7 @@
 from morpheus.pipeline.stage_schema import StageSchema
 from morpheus.pipeline.sender import Sender
 from morpheus.pipeline.receiver import Receiver
-from morpheus.pipeline.stream_wrapper import BaseStage
+from morpheus.pipeline.base_stage import BaseStage
 from morpheus.pipeline.stage import Stage
 from morpheus.pipeline.single_port_stage import SinglePortStage
 from morpheus.pipeline.multi_message_stage import MultiMessageStage

diff --git a/morpheus/pipeline/stream_wrapper.py → morpheus/pipeline/base_stage.py b/morpheus/pipeline/stream_wrapper.py → morpheus/pipeline/base_stage.py
diff --git a/morpheus/pipeline/pipeline.py b/morpheus/pipeline/pipeline.py
@@ -34,7 +34,7 @@
 from morpheus.pipeline.sender import Sender
 from morpheus.pipeline.source_stage import SourceStage
 from morpheus.pipeline.stage import Stage
-from morpheus.pipeline.stream_wrapper import BaseStage
+from morpheus.pipeline.base_stage import BaseStage
 from morpheus.utils.type_utils import pretty_print_type_name
 
 logger = logging.getLogger(__name__)

diff --git a/morpheus/pipeline/stage_schema.py b/morpheus/pipeline/stage_schema.py
@@ -16,7 +16,7 @@
 import typing
 
 if typing.TYPE_CHECKING:
-    from .stream_wrapper import BaseStage
+    from .base_stage import BaseStage
 
 
 class PortSchema: