Update ExtendLibraryFlow.py

ExtendLibraryFlow.py

@@ -8,6 +8,9 @@ class ExtendLibraryFlow(AbstractBossFlow):
         # ~~~ sets the input_data in the flow_state dict ~~~
         self._state_update_dict(update_data=input_data)
 
+        # ~~~ set the memory file to the flow state ~~~
+        self._state_update_dict(update_data={"memory_files": self.memory_files})
+
         max_rounds = self.flow_config.get("max_rounds", 1)
         if max_rounds is None:
             log.info(f"Running {self.flow_config['name']} without `max_rounds` until the early exit condition is met.")