Update agent.py
Browse files
agent.py
CHANGED
|
@@ -12,12 +12,7 @@ def run_agent(model_id, level, question, file_name, ground_truth):
|
|
| 12 |
max_steps = 10
|
| 13 |
elif level == 3:
|
| 14 |
max_steps = 20
|
| 15 |
-
|
| 16 |
-
print("###")
|
| 17 |
-
print(level)
|
| 18 |
-
print(max_steps)
|
| 19 |
-
print("###")
|
| 20 |
-
|
| 21 |
agent = CodeAgent(
|
| 22 |
add_base_tools = True,
|
| 23 |
additional_authorized_imports = ["pandas", "numpy"],
|
|
|
|
| 12 |
max_steps = 10
|
| 13 |
elif level == 3:
|
| 14 |
max_steps = 20
|
| 15 |
+
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 16 |
agent = CodeAgent(
|
| 17 |
add_base_tools = True,
|
| 18 |
additional_authorized_imports = ["pandas", "numpy"],
|