reset cache before pickle dump (#207)

LLNL · Aug 26, 2024 · 29197ea · 29197ea
1 parent cfa0b72
commit 29197ea
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/thicket/thicket.py b/thicket/thicket.py
@@ -219,6 +219,10 @@ def from_pickle(filename, **kwargs):
 
     def to_pickle(self, filename, **kwargs):
         """Write a Thicket to a pickle file."""
+
+        # Clear out statsframe to ensure thicket obj is serializable
+        self.statsframe_ops_cache = {}
+
         pickle.dump(self, open(filename, "wb"), **kwargs)
 
     @staticmethod