Spaces:
AIR-Bench
/
Running on CPU Upgrade

nan commited on
Commit
a96f80a
1 Parent(s): 7eb3ee8

chore: clean up

Browse files
Files changed (1) hide show
  1. src/benchmarks.py +3 -1
src/benchmarks.py CHANGED
@@ -113,6 +113,7 @@ class Benchmark:
113
  lang: str
114
  task: str
115
 
 
116
  qa_benchmark_dict = {}
117
  long_doc_benchmark_dict = {}
118
  for task, domain_dict in dataset_dict.items():
@@ -130,7 +131,8 @@ for task, domain_dict in dataset_dict.items():
130
  benchmark_name = get_safe_name(benchmark_name)
131
  col_name = benchmark_name
132
  for metric in metric_list:
133
- long_doc_benchmark_dict[benchmark_name] = Benchmark(benchmark_name, metric, col_name, domain, lang, task)
 
134
 
135
  BenchmarksQA = Enum('BenchmarksQA', qa_benchmark_dict)
136
  BenchmarksLongDoc = Enum('BenchmarksLongDoc', long_doc_benchmark_dict)
 
113
  lang: str
114
  task: str
115
 
116
+
117
  qa_benchmark_dict = {}
118
  long_doc_benchmark_dict = {}
119
  for task, domain_dict in dataset_dict.items():
 
131
  benchmark_name = get_safe_name(benchmark_name)
132
  col_name = benchmark_name
133
  for metric in metric_list:
134
+ long_doc_benchmark_dict[benchmark_name] = Benchmark(benchmark_name, metric, col_name, domain,
135
+ lang, task)
136
 
137
  BenchmarksQA = Enum('BenchmarksQA', qa_benchmark_dict)
138
  BenchmarksLongDoc = Enum('BenchmarksLongDoc', long_doc_benchmark_dict)