Update test_issue5230.py

This commit is contained in:
Ines Montani 2020-06-20 16:17:48 +02:00
parent 296b5d633b
commit 63c22969f4

View File

@ -76,6 +76,7 @@ def entity_linker():
# 1. no model leads to error in serialization,
# 2. the affected line is the one for model serialization
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.begin_training(pipeline=nlp.pipeline)
return entity_linker