KennyUTC commited on
Commit
26c4646
1 Parent(s): acda0b3

Update gen_table.py

Browse files
Files changed (1) hide show
  1. gen_table.py +1 -1
gen_table.py CHANGED
@@ -79,7 +79,7 @@ def BUILD_L1_DF(results, fields):
79
 
80
  def BUILD_L2_DF(results, dataset):
81
  res = defaultdict(list)
82
- fields = list(list(results.values())[0][dataset].keys())
83
  non_overall_fields = [x for x in fields if 'Overall' not in x]
84
  overall_fields = [x for x in fields if 'Overall' in x]
85
  if dataset == 'MME':
 
79
 
80
  def BUILD_L2_DF(results, dataset):
81
  res = defaultdict(list)
82
+ fields = results['LLaVA-v1.5-7B'][dataset].keys()
83
  non_overall_fields = [x for x in fields if 'Overall' not in x]
84
  overall_fields = [x for x in fields if 'Overall' in x]
85
  if dataset == 'MME':