@@ -14,7 +14,7 @@ def inspection(args):
1414 -- completion.py: prompt + completion
1515 -- execution_trace.txt: execution trace
1616 """
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" , "" ))
1818 if args .in_place :
1919 shutil .rmtree (path , ignore_errors = True )
2020 if not os .path .exists (path ):
@@ -48,12 +48,13 @@ def inspection(args):
4848 f .write ("=" * 50 + "\n " )
4949def main ():
5050 parser = argparse .ArgumentParser ()
51- parser .add_argument ("--eval-results " , required = True , type = str )
51+ parser .add_argument ("--eval_results " , required = True , type = str )
5252 parser .add_argument (
5353 "--split" , required = True , type = str , choices = ["complete" , "instruct" ]
5454 )
5555 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" )
5758 args = parser .parse_args ()
5859
5960 inspection (args )
0 commit comments