Update UpdatePlanAtomicFlow.py

UpdatePlanAtomicFlow.py

@@ -1,6 +1,7 @@
 from typing import Dict, Any
 from aiflows.base_flows.atomic import AtomicFlow
 class UpdatePlanAtomicFlow(AtomicFlow):
+    """Refer to: https://huggingface.co/Tachi67/CoderFlowModule/blob/main/UpdatePlanAtomicFlow.py"""
     def _check_input(self, input_data: Dict[str, Any]):
         assert "memory_files" in input_data, "memory_files not passed to UpdatePlanAtomicFlow"
         assert "plan" in input_data["memory_files"], "plan not in memory_files"