Skip to content

Commit f254211

Browse files
committed
fix for e2b
1 parent fcc4389 commit f254211

1 file changed

Lines changed: 4 additions & 1 deletion

File tree

bigcodebench/evaluate.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -206,7 +206,10 @@ def evaluate(
206206

207207
pass_at_k = dict()
208208

209-
passk = [int(k) for k in pass_k.split(",")]
209+
if isinstance(pass_k, str):
210+
passk = [int(k) for k in pass_k.split(",")]
211+
else:
212+
passk = pass_k
210213

211214
if parallel < 1:
212215
n_workers = max(1, multiprocessing.cpu_count() // 2)

0 commit comments

Comments
 (0)