@@ -14,7 +14,7 @@ def inspection(args):
14
14
-- completion.py: prompt + completion
15
15
-- execution_trace.txt: execution trace
16
16
"""
17
- path = os .path .join ("inspect" , args .eval_results .split ("/" )[- 1 ].replace (".json" , "" ))
17
+ path = os .path .join (args . save_path , args .eval_results .split ("/" )[- 1 ].replace (".json" , "" ))
18
18
if args .in_place :
19
19
shutil .rmtree (path , ignore_errors = True )
20
20
if not os .path .exists (path ):
@@ -48,12 +48,13 @@ def inspection(args):
48
48
f .write ("=" * 50 + "\n " )
49
49
def main ():
50
50
parser = argparse .ArgumentParser ()
51
- parser .add_argument ("--eval-results " , required = True , type = str )
51
+ parser .add_argument ("--eval_results " , required = True , type = str )
52
52
parser .add_argument (
53
53
"--split" , required = True , type = str , choices = ["complete" , "instruct" ]
54
54
)
55
55
parser .add_argument ("--subset" , default = "hard" , type = str , choices = ["full" , "hard" ])
56
- parser .add_argument ("--in-place" , action = "store_true" )
56
+ parser .add_argument ("--save_path" , default = "inspect" , type = str )
57
+ parser .add_argument ("--in_place" , action = "store_true" )
57
58
args = parser .parse_args ()
58
59
59
60
inspection (args )
0 commit comments