From ce7b51ef063fc472dae2797d99aa78c6a2d3d799 Mon Sep 17 00:00:00 2001 From: Sofie Van Landeghem Date: Tue, 2 Apr 2024 09:29:46 +0200 Subject: [PATCH] Update spacy/pipeline/entity_linker.py Co-authored-by: Raphael Mitsch --- spacy/pipeline/entity_linker.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/pipeline/entity_linker.py b/spacy/pipeline/entity_linker.py index 790baf766..03cbe4dc5 100644 --- a/spacy/pipeline/entity_linker.py +++ b/spacy/pipeline/entity_linker.py @@ -241,7 +241,7 @@ class EntityLinker(TrainablePipe): if candidates_batch_size < 1: raise ValueError(Errors.E1044) - def _score_augmented(examples, **kwargs): + def _score_augmented(examples: Iterable[Example], **kwargs): # Because of how spaCy works, we can't just score immediately, because Language.evaluate # calls pipe() on the predicted docs, which won't have entities if there is no NER in the pipeline. if not self.use_gold_ents: