From ba4850c14dfe81ff8cb42982f0c59c1ac10001ed Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Marcos=20Mart=C3=ADnez=20Galindo?= Date: Thu, 15 Aug 2024 05:59:37 +0100 Subject: [PATCH] :art: Fix Flake MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Marcos Martínez Galindo --- zshot/evaluation/dataset/med_mentions/entities.py | 2 +- zshot/evaluation/dataset/ontonotes/entities.py | 2 +- zshot/evaluation/run_evaluation.py | 2 -- 3 files changed, 2 insertions(+), 4 deletions(-) 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