Address upcoming numpy v1.25 deprecations in test suite

This commit is contained in:
Adriane Boyd 2023-05-30 15:33:22 +02:00
parent 9b7a59c325
commit 583815377e

View File

@ -72,7 +72,7 @@ def entity_linker():
def create_kb(vocab):
kb = InMemoryLookupKB(vocab, entity_vector_length=1)
kb.add_entity("test", 0.0, zeros((1, 1), dtype="f"))
kb.add_entity("test", 0.0, zeros((1,), dtype="f"))
return kb
entity_linker = nlp.add_pipe("entity_linker")