From 6ebbcf28226c2d87560277675e86518176bb8fbf Mon Sep 17 00:00:00 2001
From: Omar Elshehy <omarelshehy@Omars-MacBook-Pro.local>
Date: Mon, 9 Dec 2024 18:00:56 +0100
Subject: [PATCH] make lint changes

---
 mteb/tasks/Reranking/__init__.py                 | 2 +-
 mteb/tasks/Reranking/ara/NamaaMrTydiReranking.py | 9 ++++++---
 2 files changed, 7 insertions(+), 4 deletions(-)

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