From 9e71adc0743d123fdc19688865dd468dbdf02776 Mon Sep 17 00:00:00 2001 From: Sofie Van Landeghem Date: Sun, 19 Mar 2023 23:27:20 +0100 Subject: [PATCH] Use Candidate output type for toy generators in the test suite to mimick best practices --- spacy/tests/pipeline/test_entity_linker.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/spacy/tests/pipeline/test_entity_linker.py b/spacy/tests/pipeline/test_entity_linker.py index e29e3920b..5d2f0c430 100644 --- a/spacy/tests/pipeline/test_entity_linker.py +++ b/spacy/tests/pipeline/test_entity_linker.py @@ -510,13 +510,13 @@ def test_el_pipe_configuration(nlp): @registry.misc("spacy.LowercaseCandidateGenerator.v1") def create_candidates() -> Callable[ - [InMemoryLookupKB, "Span"], Iterable[InMemoryCandidate] + [InMemoryLookupKB, "Span"], Iterable[Candidate] ]: return get_lowercased_candidates @registry.misc("spacy.LowercaseCandidateBatchGenerator.v1") def create_candidates_batch() -> Callable[ - [InMemoryLookupKB, Iterable["Span"]], Iterable[Iterable[InMemoryCandidate]] + [InMemoryLookupKB, Iterable["Span"]], Iterable[Iterable[Candidate]] ]: return get_lowercased_candidates_batch