Update code_eval.py

code_eval.py

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