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

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (44) hide show
  1. {aspire_inference-0.1.0a3/aspire_inference.egg-info → aspire_inference-0.1.0a5}/PKG-INFO +2 -1
  2. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5/aspire_inference.egg-info}/PKG-INFO +2 -1
  3. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/aspire_inference.egg-info/requires.txt +1 -0
  4. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/pyproject.toml +1 -0
  5. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/samples.py +14 -0
  6. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/.github/workflows/lint.yml +0 -0
  7. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/.github/workflows/publish.yml +0 -0
  8. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/.github/workflows/tests.yml +0 -0
  9. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/.gitignore +0 -0
  10. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/.pre-commit-config.yaml +0 -0
  11. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/LICENSE +0 -0
  12. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/README.md +0 -0
  13. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/aspire_inference.egg-info/SOURCES.txt +0 -0
  14. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/aspire_inference.egg-info/dependency_links.txt +0 -0
  15. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/aspire_inference.egg-info/top_level.txt +0 -0
  16. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/examples/basic_example.py +0 -0
  17. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/setup.cfg +0 -0
  18. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/__init__.py +0 -0
  19. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/aspire.py +0 -0
  20. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/flows/__init__.py +0 -0
  21. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/flows/base.py +0 -0
  22. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/flows/jax/__init__.py +0 -0
  23. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/flows/jax/flows.py +0 -0
  24. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/flows/jax/utils.py +0 -0
  25. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/flows/torch/__init__.py +0 -0
  26. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/flows/torch/flows.py +0 -0
  27. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/history.py +0 -0
  28. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/plot.py +0 -0
  29. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/samplers/__init__.py +0 -0
  30. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/samplers/base.py +0 -0
  31. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/samplers/importance.py +0 -0
  32. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/samplers/mcmc.py +0 -0
  33. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/samplers/smc/__init__.py +0 -0
  34. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/samplers/smc/base.py +0 -0
  35. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/samplers/smc/blackjax.py +0 -0
  36. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/samplers/smc/emcee.py +0 -0
  37. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/samplers/smc/minipcn.py +0 -0
  38. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/transforms.py +0 -0
  39. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/utils.py +0 -0
  40. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/tests/conftest.py +0 -0
  41. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/tests/integration_tests/conftest.py +0 -0
  42. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/tests/integration_tests/test_integration.py +0 -0
  43. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/tests/test_flows/test_jax_flows/test_flowjax_flows.py +0 -0
  44. {aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/tests/test_flows/test_torch_flows/test_zuko_flows.py +0 -0
{aspire_inference-0.1.0a3/aspire_inference.egg-info → aspire_inference-0.1.0a5}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: aspire-inference
3
- Version: 0.1.0a3
3
+ Version: 0.1.0a5
4
4
  Summary: Accelerate Sequential Posterior Inference via REuse
5
5
  Author-email: "Michael J. Williams" <michaeljw1@googlemail.com>
6
6
  License: MIT
@@ -23,6 +23,7 @@ Requires-Dist: flowjax; extra == "jax"
23
23
  Provides-Extra: torch
24
24
  Requires-Dist: torch; extra == "torch"
25
25
  Requires-Dist: zuko; extra == "torch"
26
+ Requires-Dist: tqdm; extra == "torch"
26
27
  Provides-Extra: minipcn
27
28
  Requires-Dist: minipcn; extra == "minipcn"
28
29
  Provides-Extra: emcee
{aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5/aspire_inference.egg-info}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: aspire-inference
3
- Version: 0.1.0a3
3
+ Version: 0.1.0a5
4
4
  Summary: Accelerate Sequential Posterior Inference via REuse
5
5
  Author-email: "Michael J. Williams" <michaeljw1@googlemail.com>
6
6
  License: MIT
@@ -23,6 +23,7 @@ Requires-Dist: flowjax; extra == "jax"
23
23
  Provides-Extra: torch
24
24
  Requires-Dist: torch; extra == "torch"
25
25
  Requires-Dist: zuko; extra == "torch"
26
+ Requires-Dist: tqdm; extra == "torch"
26
27
  Provides-Extra: minipcn
27
28
  Requires-Dist: minipcn; extra == "minipcn"
28
29
  Provides-Extra: emcee
{aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/aspire_inference.egg-info/requires.txt RENAMED
@@ -28,3 +28,4 @@ pytest-requires
28
28
  [torch]
29
29
  torch
30
30
  zuko
31
+ tqdm
{aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/pyproject.toml RENAMED
@@ -36,6 +36,7 @@ jax = [
36
36
  torch = [
37
37
  "torch",
38
38
  "zuko",
39
+ "tqdm",
39
40
  ]
40
41
  minipcn = [
41
42
  "minipcn",
{aspire_inference-0.1.0a3 → aspire_inference-0.1.0a5}/src/aspire/samples.py RENAMED
@@ -216,6 +216,20 @@ class BaseSamples:
216
216
  **kwargs,
217
217
  )
218
218
 
219
+ def __getstate__(self):
220
+ state = self.__dict__.copy()
221
+ # replace xp (callable) with module name string
222
+ if self.xp is not None:
223
+ state["xp"] = (
224
+ self.xp.__name__ if hasattr(self.xp, "__name__") else None
225
+ )
226
+ return state
227
+
228
+ def __setstate__(self, state):
229
+ # Restore xp by checking the namespace of x
230
+ state["xp"] = array_namespace(state["x"])
231
+ self.__dict__.update(state)
232
+
219
233
 
220
234
  @dataclass
221
235
  class Samples(BaseSamples):