Add linting

embeddings-benchmark · Jul 23, 2024 · a483e3b · a483e3b
1 parent cbecd64
commit a483e3b
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/mteb/tasks/Image/ZeroshotClassification/eng/Country211.py b/mteb/tasks/Image/ZeroshotClassification/eng/Country211.py
@@ -1,7 +1,7 @@
 from __future__ import annotations
 
-import os
 import itertools
+import os
 
 from mteb.abstasks import AbsTaskZeroshotClassification
 from mteb.abstasks.TaskMetadata import TaskMetadata

diff --git a/mteb/tasks/Image/ZeroshotClassification/eng/PatchCamelyon.py b/mteb/tasks/Image/ZeroshotClassification/eng/PatchCamelyon.py
@@ -66,10 +66,10 @@ def get_candidate_labels(self) -> list[str]:
         path = os.path.dirname(__file__)
         with open(os.path.join(path, "templates/PatchCamelyon_labels.txt")) as f:
             labels = f.readlines()
-        
+
         prompts = [
             [f"a histopathology slide showing {c}" for c in labels],
-            [f"histopathology image of {c}" for c in labels]
+            [f"histopathology image of {c}" for c in labels],
         ]
 
         return list(itertools.chain(*prompts))