maryxm commited on
Commit
9db90eb
·
verified ·
1 Parent(s): c6ca0c4

Update code_eval.py

Browse files
Files changed (1) hide show
  1. 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: float(v) for k, v in pass_at_k.items()}
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