From b4c891e452623f116029a7a78d5bb1423e2d78ae Mon Sep 17 00:00:00 2001 From: Raphael Mitsch Date: Wed, 1 Mar 2023 11:32:37 +0100 Subject: [PATCH] Format. --- spacy/tests/serialize/test_serialize_kb.py | 10 +++++++--- 1 file changed, 7 insertions(+), 3 deletions(-) diff --git a/spacy/tests/serialize/test_serialize_kb.py b/spacy/tests/serialize/test_serialize_kb.py index 102abb4dc..92499e4ca 100644 --- a/spacy/tests/serialize/test_serialize_kb.py +++ b/spacy/tests/serialize/test_serialize_kb.py @@ -122,14 +122,18 @@ def test_serialize_subclassed_kb(): if not path.is_dir(): raise ValueError(Errors.E928.format(loc=path)) - def serialize_custom_fields(values: Dict[str, Any], file_path: Path) -> None: + def serialize_custom_fields( + values: Dict[str, Any], file_path: Path + ) -> None: with open(file_path, "wb") as file: pickle.dump(values, file) serialize = { "contents": lambda p: self.write_contents(p), "strings.json": lambda p: self.vocab.strings.to_disk(p), - "custom_fields": lambda p: serialize_custom_fields({"custom_field": self.custom_field}, p) + "custom_fields": lambda p: serialize_custom_fields( + {"custom_field": self.custom_field}, p + ), } util.to_disk(path, serialize, exclude) @@ -147,7 +151,7 @@ def test_serialize_subclassed_kb(): deserialize: Dict[str, Callable[[Any], Any]] = { "contents": lambda p: self.read_contents(p), "strings.json": lambda p: self.vocab.strings.from_disk(p), - "custom_fields": lambda p: deserialize_custom_fields(p) + "custom_fields": lambda p: deserialize_custom_fields(p), } util.from_disk(path, deserialize, exclude)