kennymckormick commited on
Commit
a0e2ce8
1 Parent(s): 511a860
Files changed (1) hide show
  1. gen_table.py +3 -1
gen_table.py CHANGED
@@ -79,7 +79,9 @@ 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
 
84
  non_overall_fields = [x for x in fields if 'Overall' not in x]
85
  overall_fields = [x for x in fields if 'Overall' in x]
 
79
 
80
  def BUILD_L2_DF(results, dataset):
81
  res = defaultdict(list)
82
+ sub = [v for v in results.values() if dataset in v]
83
+ assert len(sub)
84
+ fields = list(sub[dataset].keys())
85
 
86
  non_overall_fields = [x for x in fields if 'Overall' not in x]
87
  overall_fields = [x for x in fields if 'Overall' in x]