Skip to content

Commit

Permalink
Format
Browse files Browse the repository at this point in the history
  • Loading branch information
honnibal committed Oct 1, 2024
1 parent a8837be commit 725ccba
Showing 1 changed file with 9 additions and 3 deletions.
12 changes: 9 additions & 3 deletions spacy/language.py
Original file line number Diff line number Diff line change
Expand Up @@ -2143,7 +2143,9 @@ def to_disk(
serializers["tokenizer"] = lambda p: self.tokenizer.to_disk( # type: ignore[union-attr]
p, exclude=["vocab"]
)
serializers["meta.json"] = lambda p: srsly.write_json(p, _replace_numpy_floats(self.meta))
serializers["meta.json"] = lambda p: srsly.write_json(
p, _replace_numpy_floats(self.meta)
)
serializers["config.cfg"] = lambda p: self.config.to_disk(p)
for name, proc in self._components:
if name in exclude:
Expand Down Expand Up @@ -2257,7 +2259,9 @@ def to_bytes(self, *, exclude: Iterable[str] = SimpleFrozenList()) -> bytes:
serializers: Dict[str, Callable[[], bytes]] = {}
serializers["vocab"] = lambda: self.vocab.to_bytes(exclude=exclude)
serializers["tokenizer"] = lambda: self.tokenizer.to_bytes(exclude=["vocab"]) # type: ignore[union-attr]
serializers["meta.json"] = lambda: srsly.json_dumps(_replace_numpy_floats(self.meta))
serializers["meta.json"] = lambda: srsly.json_dumps(
_replace_numpy_floats(self.meta)
)
serializers["config.cfg"] = lambda: self.config.to_bytes()
for name, proc in self._components:
if name in exclude:
Expand Down Expand Up @@ -2309,7 +2313,9 @@ def deserialize_meta(b):


def _replace_numpy_floats(meta_dict: dict) -> dict:
return convert_recursive(lambda v: isinstance(v, numpy.floating), lambda v: float(v), dict(meta_dict))
return convert_recursive(
lambda v: isinstance(v, numpy.floating), lambda v: float(v), dict(meta_dict)
)


@dataclass
Expand Down

0 comments on commit 725ccba

Please sign in to comment.