Skip to content

Commit 1e24364

Browse files
committed
feat: using datasets to load
1 parent 2165431 commit 1e24364

File tree

1 file changed

+2
-8
lines changed

1 file changed

+2
-8
lines changed

bigcodebench/data/bigcodebench.py

Lines changed: 2 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -26,14 +26,8 @@ def _ready_bigcodebench_path(subset="full", version="default") -> str:
2626
)
2727

2828
extra = "-" + subset if subset != "full" else ""
29-
30-
try:
31-
dataset = load_dataset(BIGCODEBENCH_HF+extra, split=BIGCODEBENCH_VERSION)
32-
make_cache(url, dataset, path)
33-
except:
34-
if os.path.exists(path):
35-
os.remove(path)
36-
make_cache(url, None, path, gh=True)
29+
dataset = load_dataset(BIGCODEBENCH_HF+extra, split=BIGCODEBENCH_VERSION)
30+
make_cache(url, dataset, path)
3731

3832
return path
3933

0 commit comments

Comments
 (0)