diff --git a/spacy/ml/models/entity_linker.py b/spacy/ml/models/entity_linker.py index dc2009059..389add21a 100644 --- a/spacy/ml/models/entity_linker.py +++ b/spacy/ml/models/entity_linker.py @@ -138,7 +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]) + return next( + next(get_candidates_v2(kb, iter([SpanGroup(mention.doc, spans=[mention])])))[0] + ) def get_candidates_v2( diff --git a/spacy/tests/pipeline/test_entity_linker.py b/spacy/tests/pipeline/test_entity_linker.py index 955bae922..f6f6f6fd0 100644 --- a/spacy/tests/pipeline/test_entity_linker.py +++ b/spacy/tests/pipeline/test_entity_linker.py @@ -453,10 +453,13 @@ def test_candidate_generation(nlp): mykb.add_alias(alias="adam", entities=["Q2"], probabilities=[0.9]) # test the size of the relevant candidates - adam_ent_cands = next(get_candidates_v2(mykb, SpanGroup(doc=doc, spans=[adam_ent])))[0] + adam_ent_cands = next( + get_candidates_v2(mykb, SpanGroup(doc=doc, spans=[adam_ent])) + )[0] assert len(adam_ent_cands) == 1 assert ( - len(next(get_candidates_v2(mykb, SpanGroup(doc=doc, spans=[douglas_ent])))[0]) == 2 + len(next(get_candidates_v2(mykb, SpanGroup(doc=doc, spans=[douglas_ent])))[0]) + == 2 ) assert ( len(next(get_candidates_v2(mykb, SpanGroup(doc=doc, spans=[Adam_ent])))[0]) == 0