Zhejian commited on
Commit
0778ffc
·
1 Parent(s): bb41fcd
Files changed (1) hide show
  1. utils.py +3 -1
utils.py CHANGED
@@ -76,6 +76,8 @@ def parse_eval_dataset(dataset:DatasetDict) -> List[BenchmarkItem]:
76
  benchmark_items.append(BenchmarkItem(
77
  task_id=row['task_id'],
78
  question=row['question'],
79
- evaluate=EvaluateData(items=[EvaluateItem(**item) for item in row['evaluate']['items']])
 
 
80
  ))
81
  return benchmark_items
 
76
  benchmark_items.append(BenchmarkItem(
77
  task_id=row['task_id'],
78
  question=row['question'],
79
+ evaluate=EvaluateData(items=[EvaluateItem(**item) for item in row['evaluate']['items']]),
80
+ category=row['category'],
81
+ level=row['level']
82
  ))
83
  return benchmark_items