Skip to content

Commit 1dc1e37

Browse files
committed
fix
1 parent de90e7e commit 1dc1e37

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

bigcodebench/evaluate.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -214,7 +214,10 @@ def evaluate(
214214
if isinstance(selective_evaluate, str):
215215
selected_ids = set(selective_evaluate.split(","))
216216
else:
217-
selected_ids = set(selective_evaluate)
217+
try:
218+
selected_ids = set(selective_evaluate)
219+
except:
220+
selected_ids = {}
218221

219222
if parallel < 1:
220223
n_workers = max(1, multiprocessing.cpu_count() // 2)
@@ -229,10 +232,6 @@ def evaluate(
229232

230233
# Add selective evaluation logic
231234
if selective_evaluate:
232-
if isinstance(selective_evaluate, str):
233-
selected_ids = set(selective_evaluate.split(","))
234-
else:
235-
selected_ids = set(selective_evaluate)
236235
problems = {k: v for k, v in problems.items() if k in selected_ids}
237236
if not problems:
238237
raise ValueError(f"None of the provided task IDs {selected_ids} were found in the dataset")

0 commit comments

Comments
 (0)