From cfbb4a5d99297dc88aad845feabd24c8125dcf19 Mon Sep 17 00:00:00 2001 From: Raphael Mitsch Date: Mon, 24 Apr 2023 20:54:54 +0200 Subject: [PATCH] Update get_candidates() docstring. --- 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 e95f69e34..d97e3725b 100644 --- a/spacy/ml/models/entity_linker.py +++ b/spacy/ml/models/entity_linker.py @@ -118,9 +118,9 @@ def get_candidates( kb: KnowledgeBase, mentions: Iterator[SpanGroup] ) -> Iterator[Iterable[Iterable[Candidate]]]: """ - Return candidate entities for the given mentions and fetching appropriate entries from the index. + Return candidate entities for the given mentions from the KB. kb (KnowledgeBase): Knowledge base to query. - mentions (Iterator[SpanGroup]): Mentions per doc as SpanGroup instance. - RETURNS (Iterator[Iterable[Iterable[Candidate]]]): Identified candidates per document. + mentions (Iterator[SpanGroup]): Mentions per doc. + RETURNS (Iterator[Iterable[Iterable[Candidate]]]): Identified candidates per mentions in document/SpanGroup. """ return kb.get_candidates(mentions)