explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-07-04 03:43:09 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix mypy error by not using SpanGroup.__iter__()..

Browse Source
This commit is contained in:
Raphael Mitsch 2022-12-15 14:08:56 +01:00
parent b6bc6885d9
commit cdd5d69d60
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
spacy/tests/pipeline/test_entity_linker.py
View File

@ -507,7 +507,10 @@ def test_el_pipe_configuration(nlp):
kb: InMemoryLookupKB, mentions: Iterator[SpanGroup] kb: InMemoryLookupKB, mentions: Iterator[SpanGroup]
): ):
for doc_mentions in mentions: for doc_mentions in mentions:
yield [get_lowercased_candidates(kb, mention) for mention in doc_mentions] yield [
get_lowercased_candidates(kb, doc_mentions[idx])
for idx in range(len(doc_mentions))
]
@registry.misc("spacy.LowercaseCandidateGenerator.v1") @registry.misc("spacy.LowercaseCandidateGenerator.v1")
def create_candidates() -> Callable[ def create_candidates() -> Callable[