fix: change parallel logic

bigcode-project · Oct 6, 2024 · 58b3f2d · 58b3f2d
1 parent dac3a00
commit 58b3f2d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bigcodebench/evaluate.py b/bigcodebench/evaluate.py
@@ -167,7 +167,7 @@ def evaluate(
 
         pass_k = [int(k) for k in pass_k.split(",")]
 
-        if not parallel:
+        if parallel < 1:
             n_workers = max(1, multiprocessing.cpu_count() // 2)
         else:
             n_workers = parallel