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

new semantic

This commit is contained in:
Chuyue Sun 2024-06-05 13:49:34 -07:00
parent 6d60651ed1
commit a5747122cd
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
pantograph/search_llm.py
View File

@ -19,11 +19,9 @@ class LLMAgent(Agent):
"intro", "intro",
] ]
self.tactics = [ self.tactics = [
"intro h",
"cases h",
"simp", "simp",
"apply Or.inl", "rfl",
"apply Or.inr", "decide",
] ]
self.no_space_tactics = [ self.no_space_tactics = [
"assumption", "assumption",
@ -40,10 +38,8 @@ class LLMAgent(Agent):
tactics = self.tactics tactics = self.tactics
else: else:
tactics = self.no_space_tactics tactics = self.no_space_tactics
if i >= len(tactics): if i >= len(tactics):
return None return None
self.goal_tactic_id_map[key] = i + 1 self.goal_tactic_id_map[key] = i + 1
new_state = None new_state = None
for ii in range(self.n_trials): for ii in range(self.n_trials):