Update return type in EL tests.

This commit is contained in:
Raphael Mitsch 2023-03-20 09:35:50 +01:00
parent 1f2685029f
commit 48744b02b1

View File

@ -510,13 +510,13 @@ def test_el_pipe_configuration(nlp):
@registry.misc("spacy.LowercaseCandidateGenerator.v1") @registry.misc("spacy.LowercaseCandidateGenerator.v1")
def create_candidates() -> Callable[ def create_candidates() -> Callable[
[InMemoryLookupKB, "Span"], Iterable[InMemoryCandidate] [InMemoryLookupKB, "Span"], Iterable[Candidate]
]: ]:
return get_lowercased_candidates return get_lowercased_candidates
@registry.misc("spacy.LowercaseCandidateBatchGenerator.v1") @registry.misc("spacy.LowercaseCandidateBatchGenerator.v1")
def create_candidates_batch() -> Callable[ def create_candidates_batch() -> Callable[
[InMemoryLookupKB, Iterable["Span"]], Iterable[Iterable[InMemoryCandidate]] [InMemoryLookupKB, Iterable["Span"]], Iterable[Iterable[Candidate]]
]: ]:
return get_lowercased_candidates_batch return get_lowercased_candidates_batch