Skip to content

Commit d35fd70

Browse files
committed
fix(inspect): update args
1 parent 957ea7f commit d35fd70

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

bigcodebench/inspect.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -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")
4949
def 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

Comments
 (0)