Unverified Commit 9a902155 authored by Pasquale Minervini's avatar Pasquale Minervini Committed by GitHub
Browse files
parent d714fc95
...@@ -19,7 +19,7 @@ from math import exp ...@@ -19,7 +19,7 @@ from math import exp
from functools import partial from functools import partial
from packaging import version from packaging import version
from lm_eval.api.task import Task from lm_eval.api.task import ConfigurableTask
from lm_eval.api.instance import Instance from lm_eval.api.instance import Instance
_CITATION = """ _CITATION = """
...@@ -46,11 +46,14 @@ def _squad_agg(key, items): ...@@ -46,11 +46,14 @@ def _squad_agg(key, items):
return _squad_metric(predictions=predictions, references=references).get(key, 0) return _squad_metric(predictions=predictions, references=references).get(key, 0)
class SQuAD2(Task): class SQuAD2(ConfigurableTask):
VERSION = 3 VERSION = 3
DATASET_PATH = "squad_v2" DATASET_PATH = "squad_v2"
DATASET_NAME = None DATASET_NAME = None
def __init__(self):
super().__init__(config={'metadata': {'version': self.VERSION}})
# HF changed squad on us so we have to make sure we aren't running the old one # HF changed squad on us so we have to make sure we aren't running the old one
assert version.parse(datasets.__version__) >= version.parse( assert version.parse(datasets.__version__) >= version.parse(
"1.11.0" "1.11.0"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment