rbler commited on
Commit
d8bae74
·
verified ·
1 Parent(s): a25f3bd

Update evaluate.py

Browse files
Files changed (1) hide show
  1. evaluate.py +2 -1
evaluate.py CHANGED
@@ -4,7 +4,8 @@ from vg_evaluator import evaluation_for_challenge
4
  from datasets import load_dataset
5
  import os
6
  kk = load_dataset(os.getenv('VG_DATA'))['test'].to_dict()
7
-
 
8
  print(len(kk))
9
  print(kk.keys())
10
  def run_evaluation(pred_path):
 
4
  from datasets import load_dataset
5
  import os
6
  kk = load_dataset(os.getenv('VG_DATA'))['test'].to_dict()
7
+ for key in kk.keys():
8
+ kk[key] = kk[key][0]
9
  print(len(kk))
10
  print(kk.keys())
11
  def run_evaluation(pred_path):