Skip to content

Commit 4ab7c7f

Browse files
committed
fix(inspect): skip problems
1 parent a9cc5b1 commit 4ab7c7f

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

bigcodebench/inspect.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,8 @@ def inspection(args):
2828
task_path = os.path.join(path, task_id)
2929
if not os.path.exists(task_path):
3030
os.makedirs(task_path)
31+
if task_id not in problems:
32+
continue
3133
task_id_data = problems[task_id]
3234
with open(os.path.join(task_path, "ground_truth.py"), "w") as f:
3335
f.write(task_id_data[f"{args.split}_prompt"] + "\n\n" + task_id_data["canonical_solution"])

0 commit comments

Comments
 (0)