From b6d9f4c5ab7ebcf8a3845835de8b142692cba2fe Mon Sep 17 00:00:00 2001 From: Raphael Mitsch Date: Thu, 23 Feb 2023 15:14:18 +0100 Subject: [PATCH] Be more specific in KB serialization test. --- spacy/tests/serialize/test_serialize_kb.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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