From 9391de674bf2339735ec842484a38c99b11dfaee Mon Sep 17 00:00:00 2001 From: Raphael Mitsch Date: Mon, 19 Feb 2024 10:31:05 +0100 Subject: [PATCH] Fix mypy error. --- spacy/ml/models/entity_linker.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/spacy/ml/models/entity_linker.py b/spacy/ml/models/entity_linker.py index 26310ae17..a2b0a924f 100644 --- a/spacy/ml/models/entity_linker.py +++ b/spacy/ml/models/entity_linker.py @@ -138,9 +138,9 @@ def get_candidates(kb: KnowledgeBase, mention: Span) -> Iterable[Candidate]: mention (Span): Entity mention. RETURNS (Iterable[Candidate]): Identified candidates for specified mention. """ - return next( - next(get_candidates_v2(kb, iter([SpanGroup(mention.doc, spans=[mention])])))[0] # type: ignore[index] - ) + cands_per_doc = next(get_candidates_v2(kb, iter([SpanGroup(mention.doc, spans=[mention])]))) + assert isinstance(cands_per_doc, list) + return next(cands_per_doc[0]) def get_candidates_v2(