aniva/Pantograph
aniva
/
Pantograph
1
1
Fork 0
Code Issues 13 Pull Requests 4 Packages Projects Releases Wiki Activity

feat: Handle exceptions in tactic generation

This commit is contained in:
Leni Aniva 2024-10-11 22:51:07 -07:00
parent 9fc035d466
commit 8196493258
Signed by: aniva
GPG Key ID: 4D9B1C8D10EA4C50
1 changed files with 20 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
experiments/minif2f/model/llm_agent.py
View File

@ -58,6 +58,7 @@ class LLMAgent(Agent):
new_state = None new_state = None
for ii in range(self.n_trials): for ii in range(self.n_trials):
print(f"===============trail {str(ii)}============") print(f"===============trail {str(ii)}============")
try:
s = select_tactic.run( s = select_tactic.run(
server=self.server, server=self.server,
state=state, state=state,
@ -75,6 +76,9 @@ class LLMAgent(Agent):
print(colored("[Tactic] Failed:", "red"), tactic) print(colored("[Tactic] Failed:", "red"), tactic)
return None return None
return tactic return tactic
except Exception as e:
print(colored(str(e), "red"))
return None
return None return None
else: else:
self.goal_tactic_id_map[key] = i + 1 self.goal_tactic_id_map[key] = i + 1