diff --git a/spacy/tests/serialize/test_serialize_kb.py b/spacy/tests/serialize/test_serialize_kb.py index 92499e4ca..aae1423cc 100644 --- a/spacy/tests/serialize/test_serialize_kb.py +++ b/spacy/tests/serialize/test_serialize_kb.py @@ -138,6 +138,7 @@ def test_serialize_subclassed_kb(): util.to_disk(path, serialize, exclude) def from_disk(self, path, exclude: Iterable[str] = SimpleFrozenList()): + """We overwrite InMemoryLookupKB.to_disk() to ensure that self.custom_field is loaded as well.""" path = ensure_path(path) if not path.exists(): raise ValueError(Errors.E929.format(loc=path))