diff --git a/spacy/tests/serialize/test_serialize_kb.py b/spacy/tests/serialize/test_serialize_kb.py index 1309845ab..126fb5e51 100644 --- a/spacy/tests/serialize/test_serialize_kb.py +++ b/spacy/tests/serialize/test_serialize_kb.py @@ -161,4 +161,4 @@ def test_serialize_subclassed_kb(): # After IO, the KB is the standard one assert type(entity_linker2.kb) == SubInMemoryLookupKB assert entity_linker2.kb.entity_vector_length == 342 - assert hasattr(entity_linker2.kb, "custom_field") + assert entity_linker2.kb.custom_field == 666