diff --git a/examples/basic_modules/tree_textual_memory_reasoner.py b/examples/basic_modules/tree_textual_memory_reasoner.py index 42f403269..369787458 100644 --- a/examples/basic_modules/tree_textual_memory_reasoner.py +++ b/examples/basic_modules/tree_textual_memory_reasoner.py @@ -37,11 +37,18 @@ # Step 1: Prepare a mock ParsedTaskGoal parsed_goal = ParsedTaskGoal( - memories=["Multi-UAV Long-Term Coverage", "Coverage Metrics", "Reward Function Design", "Energy Model", - "CT and FT Definition", "Reward Components", "Energy Cost Components"], + memories=[ + "Multi-UAV Long-Term Coverage", + "Coverage Metrics", + "Reward Function Design", + "Energy Model", + "CT and FT Definition", + "Reward Components", + "Energy Cost Components", + ], keys=["UAV", "coverage", "energy", "reward"], tags=[], - goal_type="explanation" + goal_type="explanation", ) query = "How can multiple UAVs coordinate to maximize coverage while saving energy?"