Apply suggestions from code review

Co-authored-by: Kenneth Enevoldsen <[email protected]>
embeddings-benchmark · Dec 13, 2024 · d72b345 · d72b345
1 parent 8b1abd9
commit d72b345
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/mteb/evaluation/evaluators/Image/Any2AnyRetrievalEvaluator.py b/mteb/evaluation/evaluators/Image/Any2AnyRetrievalEvaluator.py
@@ -310,7 +310,7 @@ def __call__(
             queries,
             self.top_k,
             self.score_function,
-            task_name=self.task_name,  # type: ignore
+            task_name=self.task_name,
         )
 
     @staticmethod

diff --git a/mteb/models/align_models.py b/mteb/models/align_models.py
@@ -28,9 +28,10 @@ def __init__(
     def get_text_embeddings(
         self,
         texts: list[str],
-        batch_size: int = 32,
+        *,
         task_name: str | None = None,
         prompt_type: PromptType | None = None,
+        batch_size: int = 32,
         **kwargs: Any,
     ):
         all_text_embeddings = []