aspire-inference 0.1.0a3__py3-none-any.whl → 0.1.0a5__py3-none-any.whl

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):

{aspire_inference-0.1.0a3.dist-info → aspire_inference-0.1.0a5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: aspire-inference
-Version: 0.1.0a3
+Version: 0.1.0a5
 Summary: Accelerate Sequential Posterior Inference via REuse
 Author-email: "Michael J. Williams" <michaeljw1@googlemail.com>
 License: MIT
@@ -23,6 +23,7 @@ Requires-Dist: flowjax; extra == "jax"
 Provides-Extra: torch
 Requires-Dist: torch; extra == "torch"
 Requires-Dist: zuko; extra == "torch"
+Requires-Dist: tqdm; extra == "torch"
 Provides-Extra: minipcn
 Requires-Dist: minipcn; extra == "minipcn"
 Provides-Extra: emcee

{aspire_inference-0.1.0a3.dist-info → aspire_inference-0.1.0a5.dist-info}/RECORD

@@ -2,7 +2,7 @@ aspire/__init__.py,sha256=45R0xWaLg0aJEPK5zoTK0aIek0KOwpHwQWS1jLCDhIE,365
 aspire/aspire.py,sha256=AEkFUuOCF4F_iXUqRNst_4mucxozYRK4fG4V2wGrT4Q,15762
 aspire/history.py,sha256=l_j-riZKbTWK7Wz9zvvD_mTk9psNCKItiveYhr_pYv8,4313
 aspire/plot.py,sha256=oXwUDOb_953_ADm2KLk41JIfpE3JeiiQiSYKvUVwLqw,1423
-aspire/samples.py,sha256=hMlONOtSuYE3bU6r_wQCZ8Z1dcc3Ch15bNMLO8fGU8g,16263
+aspire/samples.py,sha256=mVyoHNcTPgqJXjKNeF2oA-KNV1asriWjToWX-T70E5k,16729
 aspire/transforms.py,sha256=fg2_UELJWjJ6gnqQi6X7s1CgKBhQ5hP7Ipil3tTjCeg,16566
 aspire/utils.py,sha256=fQeLMauCN3vAogKbVTVg9jfjW7nTEFi7V6Ot-BYNfxE,14301
 aspire/flows/__init__.py,sha256=3gGXF4HziMlZSmcEdJ_uHtrP-QEC6RXvylm4vtM-Xnk,1306
@@ -21,8 +21,8 @@ aspire/samplers/smc/base.py,sha256=GePA6tm8Dno_AjCeNuRX3KOaKnoKSFHSRAb-QWx9wJE,1
 aspire/samplers/smc/blackjax.py,sha256=9w1ORzWTT1viwp99_ttLxnNMdgTO-VqAzsf-NhgG9vY,11722
 aspire/samplers/smc/emcee.py,sha256=ZXXyN2l1Bz5ZsCPEcswg-Kakiw41nNa2jEW1N8zGjuc,2498
 aspire/samplers/smc/minipcn.py,sha256=ZjeP4iHFR67G8WKEfMe0b1McrtPgQMNHyyy4vRx6WNE,2747
-aspire_inference-0.1.0a3.dist-info/licenses/LICENSE,sha256=DN-eRtBfS9dZyT0Ds0Mdn2Y4nb-ZQ7h71vpASYBm5k4,1076
-aspire_inference-0.1.0a3.dist-info/METADATA,sha256=jwpdVn7ns4Fa6hEfFr3sNB3XXJJ5ApdKEry_H-APddA,1536
-aspire_inference-0.1.0a3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-aspire_inference-0.1.0a3.dist-info/top_level.txt,sha256=9FRIYEl2xwVFG7jSOBHsElHQ0y3_4fq01Cf4_OyMQn8,7
-aspire_inference-0.1.0a3.dist-info/RECORD,,
+aspire_inference-0.1.0a5.dist-info/licenses/LICENSE,sha256=DN-eRtBfS9dZyT0Ds0Mdn2Y4nb-ZQ7h71vpASYBm5k4,1076
+aspire_inference-0.1.0a5.dist-info/METADATA,sha256=Vq1jmMMrg6taHSFqJOCZJUex7wdakeoL7K6844VSlDs,1574
+aspire_inference-0.1.0a5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+aspire_inference-0.1.0a5.dist-info/top_level.txt,sha256=9FRIYEl2xwVFG7jSOBHsElHQ0y3_4fq01Cf4_OyMQn8,7
+aspire_inference-0.1.0a5.dist-info/RECORD,,