Spaces:
Runtime error
Runtime error
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)
|