.

code_eval.py

@@ -217,7 +217,7 @@ class CodeEval(evaluate.Metric):
                 time.sleep(1)
 
         results = {
-            task_id: [(r.completion_id, r.dict(exclude={"futures"})) for r in result]
+            task_id: [(r.prediction_id, r.dict(exclude={"futures"})) for r in result]
             for task_id, result in results.items()
         }