Fix (un)pickling

GREsau · Aug 4, 2023 · 26345d3 · 26345d3
1 parent 880753d
commit 26345d3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/localstack_persist/visitors.py b/src/localstack_persist/visitors.py
@@ -68,7 +68,7 @@ def _load_json(self, state_container: StateContainer, file_path: str):
             )
 
         unpickler = jsonpickle.Unpickler(keys=True, safe=True, on_missing="warn")
-        deserialised = unpickler.restore(envelope[DATA_KEY], reset=False)
+        deserialised = unpickler.restore(envelope[DATA_KEY])
 
         state_container.update(deserialised)
         state_container.__dict__.update(deserialised.__dict__)
@@ -90,7 +90,7 @@ def visit(self, state_container: StateContainer):
 
     def _save_json(self, state_container: dict, file_path: str):
         pickler = jsonpickle.Pickler(keys=True, warn=True)
-        flattened = pickler.flatten(state_container, reset=False)
+        flattened = pickler.flatten(state_container)
 
         envelope = {SER_VERSION_KEY: SER_VERSION, DATA_KEY: flattened}