Pretty-print JSON

borchero · Jan 2, 2022 · 78765be · 78765be
1 parent a25d5c0
commit 78765be
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/lightkit/estimator/base.py b/lightkit/estimator/base.py
@@ -161,7 +161,7 @@ def save_parameters(self, path: Path) -> None:
         """
         params = self.get_params()
         try:
-            data = json.dumps(params)
+            data = json.dumps(params, indent=4)
             with (path / "params.json").open("w+") as f:
                 f.write(data)
         except TypeError:
@@ -193,7 +193,7 @@ def save_attributes(self, path: Path) -> None:
             attribute: getattr(self, attribute) for attribute in self.persistent_attributes
         }
         try:
-            data = json.dumps(attributes)
+            data = json.dumps(attributes, indent=4)
             with (path / "attributes.json").open("w+") as f:
                 f.write(data)
         except TypeError:

diff --git a/lightkit/nn/configurable.py b/lightkit/nn/configurable.py
@@ -45,7 +45,7 @@ def save_config(self: ConfigurableModule[C], path: Path) -> None:
         """
         path.mkdir(parents=False, exist_ok=True)
         with (path / "config.json").open("w+") as f:
-            json.dump(dataclasses.asdict(self.config), f)
+            json.dump(dataclasses.asdict(self.config), f, indent=4)
 
     @jit.unused
     def save(self: ConfigurableModule[C], path: PathType, compile_model: bool = False) -> None: