fix bug in SanStream.impact_item

QSD-Group · Nov 12, 2024 · cac89b3 · cac89b3
1 parent b494632
commit cac89b3
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/qsdsan/_impact_item.py b/qsdsan/_impact_item.py
@@ -822,7 +822,7 @@ def linked_stream(self, new_s):
                          f'is replaced with {self.ID}.')
                 else:
                     warn(f'The original `StreamImpactItem` linked to stream {new_s.ID} '
-                         f'is replaced with upon the creation of a new stream.')
+                         f'is replaced upon the creation of a new stream.')
             new_s._stream_impact_item = self
         self._linked_stream = new_s
 

diff --git a/qsdsan/_sanstream.py b/qsdsan/_sanstream.py
@@ -170,12 +170,13 @@ def copy_flow(self, other, IDs=..., *, remove=False, exclude=False):
         --------
         :func:`copy` for the differences between ``copy``, ``copy_like``, and ``copy_flow``.
         '''
+        stream_impact_item = self.stream_impact_item
         Stream.copy_flow(self, other=other, IDs=IDs, remove=remove, exclude=exclude)
 
         if not isinstance(other, SanStream):
             return
 
-        self._stream_impact_item = None
+        self._stream_impact_item = stream_impact_item
 
 
     def flow_proxy(self, ID=None):