WIP: [no ci]

morpheus/pipeline/linear_pipeline.py

@@ -149,8 +149,8 @@ def add_segment_boundary(self, data_type=None, as_shared_pointer=False):
                                                       boundary_port_id=self._current_segment_id,
                                                       data_type=data_type)
 
-        # TODO: update to use data_type once typeid is attached to registered objects out of band:
-        #  https://github.com/nv-morpheus/MRC/issues/176
+        # TODO: update to use data_type once typeid is attached to registered objects out of band # pylint:disable=fixme
+        # https://github.com/nv-morpheus/MRC/issues/176
         port_id_tuple = (self._current_segment_id, object, False) if data_type else self._current_segment_id
 
         self.add_stage(boundary_egress)

morpheus/pipeline/pipeline.py

@@ -139,7 +139,7 @@ def add_edge(self,
 
         Parameters
         ----------
-        start : typing.Union[StreamWrapper, Sender]
+        start : typing.Union[BaseStage, Sender]
             The start of the edge or parent stage.
 
         end : typing.Union[Stage, Receiver]
@@ -265,7 +265,7 @@ def _pre_build(self):
     def build(self):
         """
         This function sequentially activates all the Morpheus pipeline stages passed by the users to execute a
-        pipeline. For the `Source` and all added `Stage` objects, `StreamWrapper.build` will be called sequentially to
+        pipeline. For the `Source` and all added `Stage` objects, `BaseStage.build` will be called sequentially to
         construct the pipeline.
 
         Once the pipeline has been constructed, this will start the pipeline by calling `Source.start` on the source

morpheus/pipeline/receiver.py

@@ -25,12 +25,12 @@
 
 class Receiver():
     """
-    The `Receiver` object represents a downstream port on a `StreamWrapper` object that gets messages from a `Sender`.
+    The `Receiver` object represents a downstream port on a `BaseStage` object that gets messages from a `Sender`.
 
     Parameters
         ----------
-        parent : `morpheus.pipeline.pipeline.StreamWrapper`
-            Parent `StreamWrapper` object.
+        parent : `morpheus.pipeline.pipeline.BaseStage`
+            Parent `BaseStage` object.
         port_number : int
             Receiver port number.
     """

morpheus/pipeline/sender.py

@@ -24,12 +24,12 @@
 
 class Sender():
     """
-    The `Sender` object represents a port on a `StreamWrapper` object that sends messages to a `Receiver`.
+    The `Sender` object represents a port on a `BaseStage` object that sends messages to a `Receiver`.
 
     Parameters
         ----------
-        parent : `morpheus.pipeline.pipeline.StreamWrapper`
-            Parent `StreamWrapper` object.
+        parent : `morpheus.pipeline.pipeline.BaseStage`
+            Parent `BaseStage` object.
         port_number : int
             Sender port number.
     """

morpheus/pipeline/stream_wrapper.py

@@ -234,7 +234,7 @@ def get_all_input_stages(self) -> list["BaseStage"]:
 
         Returns
         -------
-        list[`morpheus.pipeline.pipeline.StreamWrapper`]
+        list[`morpheus.pipeline.pipeline.BaseStage`]
             All input stages.
         """
         return [x.parent for x in self.get_all_inputs()]
@@ -261,7 +261,7 @@ def get_all_output_stages(self) -> list["BaseStage"]:
 
         Returns
         -------
-        list[`morpheus.pipeline.pipeline.StreamWrapper`]
+        list[`morpheus.pipeline.pipeline.BaseStage`]
             All output stages.
         """
         return [x.parent for x in self.get_all_outputs()]