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 |
|