From 79798c018163293a21fc44f63454290f9d8167e6 Mon Sep 17 00:00:00 2001 From: Raphael Mitsch Date: Mon, 19 Feb 2024 10:25:35 +0100 Subject: [PATCH] Format. --- spacy/ml/models/entity_linker.py | 4 +++- spacy/tests/pipeline/test_entity_linker.py | 7 +++++-- 2 files changed, 8 insertions(+), 3 deletions(-) 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