rbler commited on
Commit
70cd540
·
verified ·
1 Parent(s): baf7a30

Update evaluate.py

Browse files
Files changed (1) hide show
  1. evaluate.py +1 -1
evaluate.py CHANGED
@@ -3,7 +3,7 @@ import json
3
  from vg_evaluator import evaluation_for_challenge
4
  from datasets import load_dataset
5
  import os
6
- kk = load_dataset(os.getenv('VG_DATA'))
7
 
8
  print(len(kk))
9
  print(kk.keys())
 
3
  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']
7
 
8
  print(len(kk))
9
  print(kk.keys())