fix bug

liaojiajia · liaojiajia · commit 560c1e0fce62 · 2025-02-14T14:01:08.000+08:00
diff --git a/main.py b/main.py
@@ -358,7 +358,7 @@ def parse_arguments():
         "--agent",
         type=str,
         default="cot",
-        choices=["cot", "pot", "sc_cot", "react", "dnc"],
+        choices=["cot", "pot", "sc_cot", "react", "dnc", "tot"],
         help="agent used for experiment",
     )
     parser.add_argument("--system_prompt", type=str, default="", help="system prompt")
diff --git a/requirements.txt b/requirements.txt
@@ -1,5 +1,2 @@
 argparse
-json
-os
-datetime
 

Original file line number	Diff line number	Diff line change
`@@ -358,7 +358,7 @@ def parse_arguments():`
`358`	`358`	`"--agent",`
`359`	`359`	`type=str,`
`360`	`360`	`default="cot",`
`361`		`- choices=["cot", "pot", "sc_cot", "react", "dnc"],`
	`361`	`+ choices=["cot", "pot", "sc_cot", "react", "dnc", "tot"],`
`362`	`362`	`help="agent used for experiment",`
`363`	`363`	`)`
`364`	`364`	`parser.add_argument("--system_prompt", type=str, default="", help="system prompt")`
-Original file line number
+Diff line change
@@ @@ -1,5 +1,2 @@ @@
 argparse
 -json
 -os
 -datetime