Fix mypy error.

This commit is contained in:
Raphael Mitsch 2024-02-19 10:31:05 +01:00
parent c187b13d2b
commit 9391de674b

View File

@ -138,9 +138,9 @@ def get_candidates(kb: KnowledgeBase, mention: Span) -> Iterable[Candidate]:
mention (Span): Entity mention. mention (Span): Entity mention.
RETURNS (Iterable[Candidate]): Identified candidates for specified mention. RETURNS (Iterable[Candidate]): Identified candidates for specified mention.
""" """
return next( cands_per_doc = next(get_candidates_v2(kb, iter([SpanGroup(mention.doc, spans=[mention])])))
next(get_candidates_v2(kb, iter([SpanGroup(mention.doc, spans=[mention])])))[0] # type: ignore[index] assert isinstance(cands_per_doc, list)
) return next(cands_per_doc[0])
def get_candidates_v2( def get_candidates_v2(