Update code_eval.py

code_eval.py

@@ -190,7 +190,8 @@ class CodeEval(evaluate.Metric):
         correct = np.array(correct)
 
         ks = k
-        pass_at_k = {f"pass@{k}": estimate_pass_at_k(total, correct, k).mean().item() for k in ks if (total >= k).all()}
+        pass_at_k = {f"pass@{k}": estimate_pass_at_k(total, correct, k).mean() for k in ks if (total >= k).all()}
+        pass_at_k = {k: float(v) for k, v in pass_at_k.items()}
 
         return pass_at_k, results