🎨 Fix Flake

Signed-off-by: Marcos Martínez Galindo <[email protected]>
IBM · Aug 15, 2024 · ba4850c · ba4850c
1 parent de07078
commit ba4850c
Show file tree

Hide file tree

Showing 3 changed files with 2 additions and 4 deletions.
diff --git a/zshot/evaluation/dataset/med_mentions/entities.py b/zshot/evaluation/dataset/med_mentions/entities.py
@@ -15,7 +15,7 @@
 }
 
 MEDMENTIONS_EXPLANATORY_INVERSE_MAPPING = {
-    v:k for k, v in MEDMENTIONS_EXPLANATORY_MAPPING.items()
+    v: k for k, v in MEDMENTIONS_EXPLANATORY_MAPPING.items()
 }
 
 MEDMENTIONS_SPLITS = {

diff --git a/zshot/evaluation/dataset/ontonotes/entities.py b/zshot/evaluation/dataset/ontonotes/entities.py
@@ -10,7 +10,7 @@
 }
 
 ONTONOTES_EXPLANATORY_INVERSE_MAPPING = {
-    v:k for k, v in ONTONOTES_EXPLANATORY_MAPPING.items()
+    v: k for k, v in ONTONOTES_EXPLANATORY_MAPPING.items()
 }
 
 ONTONOTES_ENTITIES = [Entity(name='NEG',

diff --git a/zshot/evaluation/run_evaluation.py b/zshot/evaluation/run_evaluation.py
@@ -1,8 +1,6 @@
 import argparse
-from typing import Union
 
 import spacy
-from datasets import DatasetDict
 
 from zshot import PipelineConfig
 from zshot.evaluation import load_medmentions_zs, load_ontonotes_zs