diff --git a/zshot/evaluation/dataset/med_mentions/entities.py b/zshot/evaluation/dataset/med_mentions/entities.py
index 81e6f57..2b3615d 100644
--- 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
index e95640a..d7acf69 100644
--- 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
index 914843e..773744f 100644
--- 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