Update agent.py

agent.py

@@ -10,9 +10,14 @@ def run_agent(model_id, level, question, file_name):
         max_steps = 5
     elif level == 2:
         max_steps = 10
-    elif level == 3:
+    else:
         max_steps = 20
-        
+
+    print("###")
+    print(level)
+    print(max_steps)
+    print("###")
+    
     agent = CodeAgent(
             add_base_tools=True,
             additional_authorized_imports=["pandas", "numpy"],