jjkim commited on
Commit
e74e764
·
1 Parent(s): b446f13

fix minor index error

Browse files
Files changed (1) hide show
  1. code_eval.py +3 -3
code_eval.py CHANGED
@@ -203,9 +203,9 @@ class CodeEval(evaluate.Metric):
203
  group = list(group)
204
  new_result.append(
205
  dict(
206
- task_id=group[0]["task_id"],
207
- passed=all(r["passed"] for r in group),
208
- result=[r["result"] for r in group],
209
  completion_id=completion_id,
210
  )
211
  )
 
203
  group = list(group)
204
  new_result.append(
205
  dict(
206
+ task_id=group[0][0],
207
+ passed=all(r[1]["passed"] for r in group),
208
+ result=[r[1]["result"] for r in group],
209
  completion_id=completion_id,
210
  )
211
  )