Update UpdatePlanAtomicFlow.py

UpdatePlanAtomicFlow.py

@@ -2,6 +2,10 @@
 from typing import Dict, Any
 from aiflows.base_flows.atomic import AtomicFlow
 class UpdatePlanAtomicFlow(AtomicFlow):
+    """This class is used to update the plan file with the updated plan, called by the controlller,
+    when it realizes one step of the plan is done, and provide the updated plan, it is exactly the same
+    as the old plan, except the step that is done is marked as done.
+    """
     def _check_input(self, input_data: Dict[str, Any]):
         assert "memory_files" in input_data, "memory_files not passed to UpdatePlanAtomicFlow.yaml"
         assert "plan" in input_data["memory_files"], "plan not in memory_files"