Spaces:
Running
Running
Update Space (evaluate main: e4a27243)
Browse files- requirements.txt +1 -1
- trec_eval.py +4 -1
requirements.txt
CHANGED
|
@@ -1,2 +1,2 @@
|
|
| 1 |
-
git+https://github.com/huggingface/evaluate@
|
| 2 |
trectools
|
|
|
|
| 1 |
+
git+https://github.com/huggingface/evaluate@e4a2724377909fe2aeb4357e3971e5a569673b39
|
| 2 |
trectools
|
trec_eval.py
CHANGED
|
@@ -71,12 +71,15 @@ Examples:
|
|
| 71 |
class TRECEval(evaluate.Metric):
|
| 72 |
"""Compute TREC evaluation scores."""
|
| 73 |
|
| 74 |
-
|
|
|
|
|
|
|
| 75 |
return evaluate.MetricInfo(
|
| 76 |
module_type="metric",
|
| 77 |
description=_DESCRIPTION,
|
| 78 |
citation=_CITATION,
|
| 79 |
inputs_description=_KWARGS_DESCRIPTION,
|
|
|
|
| 80 |
features=datasets.Features(
|
| 81 |
{
|
| 82 |
"predictions": {
|
|
|
|
| 71 |
class TRECEval(evaluate.Metric):
|
| 72 |
"""Compute TREC evaluation scores."""
|
| 73 |
|
| 74 |
+
ALLOWED_CONFIG_NAMES = ["default"]
|
| 75 |
+
|
| 76 |
+
def _info(self, config):
|
| 77 |
return evaluate.MetricInfo(
|
| 78 |
module_type="metric",
|
| 79 |
description=_DESCRIPTION,
|
| 80 |
citation=_CITATION,
|
| 81 |
inputs_description=_KWARGS_DESCRIPTION,
|
| 82 |
+
config=config,
|
| 83 |
features=datasets.Features(
|
| 84 |
{
|
| 85 |
"predictions": {
|