Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
chore: clean up
Browse files- 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,
|
|
|
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)
|