diff --git a/spacy/ml/models/entity_linker.py b/spacy/ml/models/entity_linker.py index 389add21a..26310ae17 100644 --- a/spacy/ml/models/entity_linker.py +++ b/spacy/ml/models/entity_linker.py @@ -139,7 +139,7 @@ def get_candidates(kb: KnowledgeBase, mention: Span) -> Iterable[Candidate]: RETURNS (Iterable[Candidate]): Identified candidates for specified mention. """ return next( - next(get_candidates_v2(kb, iter([SpanGroup(mention.doc, spans=[mention])])))[0] + next(get_candidates_v2(kb, iter([SpanGroup(mention.doc, spans=[mention])])))[0] # type: ignore[index] )