aspire-inference 0.1.0a6__tar.gz → 0.1.0a7__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 (48) hide show
  1. {aspire_inference-0.1.0a6/aspire_inference.egg-info → aspire_inference-0.1.0a7}/PKG-INFO +1 -1
  2. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7/aspire_inference.egg-info}/PKG-INFO +1 -1
  3. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/samples.py +20 -2
  4. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/.github/workflows/lint.yml +0 -0
  5. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/.github/workflows/publish.yml +0 -0
  6. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/.github/workflows/tests.yml +0 -0
  7. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/.gitignore +0 -0
  8. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/.pre-commit-config.yaml +0 -0
  9. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/LICENSE +0 -0
  10. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/README.md +0 -0
  11. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/aspire_inference.egg-info/SOURCES.txt +0 -0
  12. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/aspire_inference.egg-info/dependency_links.txt +0 -0
  13. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/aspire_inference.egg-info/requires.txt +0 -0
  14. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/aspire_inference.egg-info/top_level.txt +0 -0
  15. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/examples/basic_example.py +0 -0
  16. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/examples/smc_example.py +0 -0
  17. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/pyproject.toml +0 -0
  18. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/setup.cfg +0 -0
  19. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/__init__.py +0 -0
  20. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/aspire.py +0 -0
  21. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/flows/__init__.py +0 -0
  22. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/flows/base.py +0 -0
  23. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/flows/jax/__init__.py +0 -0
  24. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/flows/jax/flows.py +0 -0
  25. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/flows/jax/utils.py +0 -0
  26. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/flows/torch/__init__.py +0 -0
  27. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/flows/torch/flows.py +0 -0
  28. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/history.py +0 -0
  29. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/plot.py +0 -0
  30. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/samplers/__init__.py +0 -0
  31. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/samplers/base.py +0 -0
  32. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/samplers/importance.py +0 -0
  33. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/samplers/mcmc.py +0 -0
  34. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/samplers/smc/__init__.py +0 -0
  35. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/samplers/smc/base.py +0 -0
  36. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/samplers/smc/blackjax.py +0 -0
  37. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/samplers/smc/emcee.py +0 -0
  38. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/samplers/smc/minipcn.py +0 -0
  39. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/transforms.py +0 -0
  40. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/utils.py +0 -0
  41. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/tests/conftest.py +0 -0
  42. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/tests/integration_tests/conftest.py +0 -0
  43. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/tests/integration_tests/test_integration.py +0 -0
  44. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/tests/test_flows/test_jax_flows/test_flowjax_flows.py +0 -0
  45. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/tests/test_flows/test_torch_flows/test_zuko_flows.py +0 -0
  46. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/tests/test_samples.py +0 -0
  47. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/tests/test_transforms.py +0 -0
  48. {aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/tests/test_utils.py +0 -0
{aspire_inference-0.1.0a6/aspire_inference.egg-info → aspire_inference-0.1.0a7}/PKG-INFO RENAMED
@@ -1,6 +1,6 @@
1
1
  Metadata-Version: 2.4
2
2
  Name: aspire-inference
3
- Version: 0.1.0a6
3
+ Version: 0.1.0a7
4
4
  Summary: Accelerate Sequential Posterior Inference via REuse
5
5
  Author-email: "Michael J. Williams" <michaeljw1@googlemail.com>
6
6
  License: MIT
{aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7/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.0a6
3
+ Version: 0.1.0a7
4
4
  Summary: Accelerate Sequential Posterior Inference via REuse
5
5
  Author-email: "Michael J. Williams" <michaeljw1@googlemail.com>
6
6
  License: MIT
{aspire_inference-0.1.0a6 → aspire_inference-0.1.0a7}/src/aspire/samples.py RENAMED
@@ -14,6 +14,7 @@ from array_api_compat import (
14
14
  )
15
15
  from array_api_compat import device as api_device
16
16
  from array_api_compat.common._typing import Array
17
+ from matplotlib.figure import Figure
17
18
 
18
19
  from .utils import (
19
20
  asarray,
@@ -161,7 +162,24 @@ class BaseSamples:
161
162
 
162
163
  return pd.DataFrame(self.to_dict(flat=flat))
163
164
 
164
- def plot_corner(self, parameters: list[str] | None = None, **kwargs):
165
+ def plot_corner(
166
+ self,
167
+ parameters: list[str] | None = None,
168
+ fig: Figure | None = None,
169
+ **kwargs,
170
+ ):
171
+ """Plot a corner plot of the samples.
172
+
173
+ Parameters
174
+ ----------
175
+ parameters : list[str] | None
176
+ List of parameters to plot. If None, all parameters are plotted.
177
+ fig : matplotlib.figure.Figure | None
178
+ Figure to plot on. If None, a new figure is created.
179
+ **kwargs : dict
180
+ Additional keyword arguments to pass to corner.corner(). Kwargs
181
+ are deep-copied before use.
182
+ """
165
183
  import corner
166
184
 
167
185
  kwargs = copy.deepcopy(kwargs)
@@ -173,7 +191,7 @@ class BaseSamples:
173
191
  x = self.x[:, indices] if self.x.ndim > 1 else self.x[indices]
174
192
  else:
175
193
  x = self.x
176
- fig = corner.corner(to_numpy(x), **kwargs)
194
+ fig = corner.corner(to_numpy(x), fig=fig, **kwargs)
177
195
  return fig
178
196
 
179
197
  def __str__(self):