explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-05 21:00:19 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix or ignore mypy issues in test.

Browse Source
This commit is contained in:
Raphael Mitsch 2023-03-13 10:57:00 +01:00
parent 8729dba619
commit b324655afb
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -150,15 +150,14 @@ def test_sentence_crossing_ents(entity_in_first_sentence: bool):
# Create the Entity Linker component and add it to the pipeline
entity_linker = nlp.add_pipe("entity_linker", last=True)
entity_linker.set_kb(create_kb)
entity_linker.set_kb(create_kb) # type: ignore
# train the NEL pipe
optimizer = nlp.initialize(get_examples=lambda: train_examples)
for i in range(2):
losses = {}
nlp.update(train_examples, sgd=optimizer, losses=losses)
nlp.update(train_examples, sgd=optimizer)
# This shouldn't crash.
entity_linker.predict([example.reference])
entity_linker.predict([example.reference]) # type: ignore
def test_no_entities():