Update code_eval.py
Browse files- code_eval.py +1 -1
code_eval.py
CHANGED
@@ -191,7 +191,7 @@ class CodeEval(evaluate.Metric):
|
|
191 |
|
192 |
ks = k
|
193 |
pass_at_k = {f"pass@{k}": estimate_pass_at_k(total, correct, k).mean() for k in ks if (total >= k).all()}
|
194 |
-
pass_at_k = {k:
|
195 |
|
196 |
return pass_at_k, results
|
197 |
|
|
|
191 |
|
192 |
ks = k
|
193 |
pass_at_k = {f"pass@{k}": estimate_pass_at_k(total, correct, k).mean() for k in ks if (total >= k).all()}
|
194 |
+
pass_at_k = {k: str(v.item()) for k, v in pass_at_k.items()}
|
195 |
|
196 |
return pass_at_k, results
|
197 |
|