diff --git a/rsi/rsi.py b/rsi/rsi.py index ed3c3a6..6764146 100644 --- a/rsi/rsi.py +++ b/rsi/rsi.py @@ -28,7 +28,7 @@ def new_state(self, directions: int, name: str) -> State: self.set_state(newstate, name) return newstate - def write(self, path: Union[str, Path], make_parent_dirs: bool = True) -> None: + def write(self, path: Union[str, Path], make_parent_dirs: bool = True, indent: Optional[int] = None) -> None: if isinstance(path, str): path = Path(path) @@ -67,7 +67,7 @@ def write(self, path: Union[str, Path], make_parent_dirs: bool = True) -> None: metajson["states"] = states with metapath.open("w") as f: - f.write(json.dumps(metajson)) + f.write(json.dumps(metajson, indent=indent)) # Write PNG files. for state in self.states.values():