aspire-inference 0.1.0a4__tar.gz → 0.1.0a5__tar.gz

{aspire_inference-0.1.0a4/aspire_inference.egg-info → aspire_inference-0.1.0a5}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aspire-inference
-Version: 0.1.0a4
+Version: 0.1.0a5
 Summary: Accelerate Sequential Posterior Inference via REuse
 Author-email: "Michael J. Williams" <michaeljw1@googlemail.com>
 License: MIT

{aspire_inference-0.1.0a4 → aspire_inference-0.1.0a5/aspire_inference.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aspire-inference
-Version: 0.1.0a4
+Version: 0.1.0a5
 Summary: Accelerate Sequential Posterior Inference via REuse
 Author-email: "Michael J. Williams" <michaeljw1@googlemail.com>
 License: MIT

{aspire_inference-0.1.0a4 → aspire_inference-0.1.0a5}/src/aspire/samples.py

@@ -216,6 +216,20 @@ class BaseSamples:
             **kwargs,
         )
 
+    def __getstate__(self):
+        state = self.__dict__.copy()
+        # replace xp (callable) with module name string
+        if self.xp is not None:
+            state["xp"] = (
+                self.xp.__name__ if hasattr(self.xp, "__name__") else None
+            )
+        return state
+
+    def __setstate__(self, state):
+        # Restore xp by checking the namespace of x
+        state["xp"] = array_namespace(state["x"])
+        self.__dict__.update(state)
+
 
 @dataclass
 class Samples(BaseSamples):