diff --git a/mteb/tasks/Reranking/__init__.py b/mteb/tasks/Reranking/__init__.py index 576ba8c72..2c3a27919 100644 --- a/mteb/tasks/Reranking/__init__.py +++ b/mteb/tasks/Reranking/__init__.py @@ -1,5 +1,6 @@ from __future__ import annotations +from .ara.NamaaMrTydiReranking import * from .eng.AskUbuntuDupQuestions import * from .eng.MindSmallReranking import * from .eng.SciDocsReranking import * @@ -13,4 +14,3 @@ from .multilingual.WikipediaRerankingMultilingual import * from .rus.RuBQReranking import * from .zho.CMTEBReranking import * -from .ara.NamaaMrTydiReranking import * diff --git a/mteb/tasks/Reranking/ara/NamaaMrTydiReranking.py b/mteb/tasks/Reranking/ara/NamaaMrTydiReranking.py index 0ea89282b..130d5c90d 100644 --- a/mteb/tasks/Reranking/ara/NamaaMrTydiReranking.py +++ b/mteb/tasks/Reranking/ara/NamaaMrTydiReranking.py @@ -1,8 +1,11 @@ +from __future__ import annotations + +import datasets + from mteb.abstasks.TaskMetadata import TaskMetadata from ....abstasks.AbsTaskReranking import AbsTaskReranking -import datasets class NamaaMrTydiReranking(AbsTaskReranking): metadata = TaskMetadata( @@ -42,7 +45,7 @@ def load_data(self, **kwargs): self.dataset = self.dataset.map( lambda x: { - "query" : x["query"], + "query": x["query"], "positive": [x["positive"]], "negative": x["negative"], } @@ -50,4 +53,4 @@ def load_data(self, **kwargs): self.dataset = datasets.DatasetDict({"test": self.dataset}) self.dataset_transform() - self.data_loaded = True \ No newline at end of file + self.data_loaded = True