make lint changes

embeddings-benchmark · Dec 9, 2024 · 6ebbcf2 · 6ebbcf2
1 parent df3d18d
commit 6ebbcf2
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 4 deletions.
diff --git 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
@@ -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,12 +45,12 @@ def load_data(self, **kwargs):
 
         self.dataset = self.dataset.map(
             lambda x: {
-                "query" : x["query"],
+                "query": x["query"],
                 "positive": [x["positive"]],
                 "negative": x["negative"],
             }
         )
         self.dataset = datasets.DatasetDict({"test": self.dataset})
         self.dataset_transform()
 
-        self.data_loaded = True
+        self.data_loaded = True