Update app.py

app.py

@@ -65,10 +65,6 @@ class MasterAgent:
             print(f"Error instantiating agent: {e}")
             return f"Error initializing agent: {e}", None
 
-        # Seems like this format was updated, we'll use the new default for now.
-        #with open("prompts.yaml", 'r') as stream:
-        #    prompt_templates = yaml.safe_load(stream)
-
         print("Starting instantiation of master")
         self.master_agent = CodeAgent(
             tools=[
@@ -80,7 +76,6 @@ class MasterAgent:
             add_base_tools=True,
             planning_interval=5,
             max_steps=15,
-            #prompt_templates=prompt_templates,
             managed_agents=[coder.agent],
             additional_authorized_imports=[
                 "geopandas",
@@ -223,8 +218,8 @@ def run_and_submit_all( profile: gr.OAuthProfile | None):
             answers_payload.append({"task_id": task_id, "submitted_answer": submitted_answer})
             results_log.append({"Task ID": task_id, "Question": question_text, "Submitted Answer": submitted_answer})
 
-            with open(task_answer_path, 'w'):
-                yaml.dump({"task_id": task_id, "question_text": question_text, "submitted_answer": submitted_answer})
+            with open(task_answer_path, 'w') as file:
+                yaml.dump({"task_id": task_id, "question_text": question_text, "submitted_answer": submitted_answer}, file)
             
         except Exception as e:
              print(f"Error running agent on task {task_id}: {e}")