explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-26 01:46:28 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update test_issue5230.py

Browse Source
This commit is contained in:
Ines Montani 2020-06-20 16:17:48 +02:00
parent 296b5d633b
commit 63c22969f4
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
spacy/tests/regression/test_issue5230.py
View File

@ -76,6 +76,7 @@ def entity_linker():
# 1. no model leads to error in serialization, # 1. no model leads to error in serialization,
# 2. the affected line is the one for model serialization # 2. the affected line is the one for model serialization
kb = KnowledgeBase(nlp.vocab, entity_vector_length=1) kb = KnowledgeBase(nlp.vocab, entity_vector_length=1)
kb.add_entity("test", 0.0, zeros((1, 1), dtype="f"))
entity_linker.set_kb(kb) entity_linker.set_kb(kb)
entity_linker.begin_training(pipeline=nlp.pipeline) entity_linker.begin_training(pipeline=nlp.pipeline)
return entity_linker return entity_linker