From 5a9d8ba73cbc1ddd7c901723849614a7b14a8cf3 Mon Sep 17 00:00:00 2001 From: Raphael Mitsch Date: Tue, 28 Feb 2023 13:56:13 +0100 Subject: [PATCH] Format. --- spacy/ml/models/entity_linker.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/spacy/ml/models/entity_linker.py b/spacy/ml/models/entity_linker.py index df42c09dc..165302c3b 100644 --- a/spacy/ml/models/entity_linker.py +++ b/spacy/ml/models/entity_linker.py @@ -121,7 +121,9 @@ def get_candidates(kb: KnowledgeBase, mention: Span) -> Iterable[Candidate]: return kb.get_candidates(mention) -def get_candidates_batch(kb: KnowledgeBase, mentions: Iterable[Span]) -> Iterable[Iterable[Candidate]]: +def get_candidates_batch( + kb: KnowledgeBase, mentions: Iterable[Span] +) -> Iterable[Iterable[Candidate]]: """ Return candidate entities for the given mentions and fetching appropriate entries from the index. kb (KnowledgeBase): Knowledge base to query.