@@ -31,3 +30,15 @@ class HeadQA(HFTask, MultipleChoiceTask):
defdoc_to_text(self,doc):
returndoc["query"]
classHeadQAEn(HeadQABase):
DATASET_NAME="en"
classHeadQAEs(HeadQABase):
DATASET_NAME="es"
# for backwards compatibility
classHeadQAEsDeprecated(HeadQABase):
DATASET_NAME="es"
print("WARNING: headqa is deprecated. Please use headqa_es or headqa_en instead. See https://github.com/EleutherAI/lm-evaluation-harness/pull/240 for more info.")