arviz 0.21.0__py3-none-any.whl → 0.22.0__py3-none-any.whl

arviz/tests/external_tests/test_data_numpyro.py

@@ -46,7 +46,9 @@ class TestDataNumPyro:
         )
         return predictions
 
-    def get_inference_data(self, data, eight_schools_params, predictions_data, predictions_params):
+    def get_inference_data(
+        self, data, eight_schools_params, predictions_data, predictions_params, infer_dims=False
+    ):
         posterior_samples = data.obj.get_samples()
         model = data.obj.sampler.model
         posterior_predictive = Predictive(model, posterior_samples)(
@@ -55,6 +57,12 @@ class TestDataNumPyro:
         prior = Predictive(model, num_samples=500)(
             PRNGKey(2), eight_schools_params["J"], eight_schools_params["sigma"]
         )
+        dims = {"theta": ["school"], "eta": ["school"], "obs": ["school"]}
+        pred_dims = {"theta": ["school_pred"], "eta": ["school_pred"], "obs": ["school_pred"]}
+        if infer_dims:
+            dims = None
+            pred_dims = None
+
         predictions = predictions_data
         return from_numpyro(
             posterior=data.obj,
@@ -65,8 +73,8 @@ class TestDataNumPyro:
                 "school": np.arange(eight_schools_params["J"]),
                 "school_pred": np.arange(predictions_params["J"]),
             },
-            dims={"theta": ["school"], "eta": ["school"], "obs": ["school"]},
-            pred_dims={"theta": ["school_pred"], "eta": ["school_pred"], "obs": ["school_pred"]},
+            dims=dims,
+            pred_dims=pred_dims,
         )
 
     def test_inference_data_namedtuple(self, data):
@@ -77,6 +85,7 @@ class TestDataNumPyro:
         data.obj.get_samples = lambda *args, **kwargs: data_namedtuple
         inference_data = from_numpyro(
             posterior=data.obj,
+            dims={},  # This mock test needs to turn off autodims like so or mock group_by_chain
         )
         assert isinstance(data.obj.get_samples(), Samples)
         data.obj.get_samples = _old_fn
@@ -282,3 +291,121 @@ class TestDataNumPyro:
         mcmc.run(PRNGKey(0))
         inference_data = from_numpyro(mcmc)
         assert inference_data.observed_data
+
+    def test_mcmc_infer_dims(self):
+        import numpyro
+        import numpyro.distributions as dist
+        from numpyro.infer import MCMC, NUTS
+
+        def model():
+            # note: group2 gets assigned dim=-1 and group1 is assigned dim=-2
+            with numpyro.plate("group2", 5), numpyro.plate("group1", 10):
+                _ = numpyro.sample("param", dist.Normal(0, 1))
+
+        mcmc = MCMC(NUTS(model), num_warmup=10, num_samples=10)
+        mcmc.run(PRNGKey(0))
+        inference_data = from_numpyro(
+            mcmc, coords={"group1": np.arange(10), "group2": np.arange(5)}
+        )
+        assert inference_data.posterior.param.dims == ("chain", "draw", "group1", "group2")
+        assert all(dim in inference_data.posterior.param.coords for dim in ("group1", "group2"))
+
+    def test_mcmc_infer_unsorted_dims(self):
+        import numpyro
+        import numpyro.distributions as dist
+        from numpyro.infer import MCMC, NUTS
+
+        def model():
+            group1_plate = numpyro.plate("group1", 10, dim=-1)
+            group2_plate = numpyro.plate("group2", 5, dim=-2)
+
+            # the plate contexts are entered in a different order than the pre-defined dims
+            # we should make sure this still works because the trace has all of the info it needs
+            with group2_plate, group1_plate:
+                _ = numpyro.sample("param", dist.Normal(0, 1))
+
+        mcmc = MCMC(NUTS(model), num_warmup=10, num_samples=10)
+        mcmc.run(PRNGKey(0))
+        inference_data = from_numpyro(
+            mcmc, coords={"group1": np.arange(10), "group2": np.arange(5)}
+        )
+        assert inference_data.posterior.param.dims == ("chain", "draw", "group2", "group1")
+        assert all(dim in inference_data.posterior.param.coords for dim in ("group1", "group2"))
+
+    def test_mcmc_infer_dims_no_coords(self):
+        import numpyro
+        import numpyro.distributions as dist
+        from numpyro.infer import MCMC, NUTS
+
+        def model():
+            with numpyro.plate("group", 5):
+                _ = numpyro.sample("param", dist.Normal(0, 1))
+
+        mcmc = MCMC(NUTS(model), num_warmup=10, num_samples=10)
+        mcmc.run(PRNGKey(0))
+        inference_data = from_numpyro(mcmc)
+        assert inference_data.posterior.param.dims == ("chain", "draw", "group")
+
+    def test_mcmc_event_dims(self):
+        import numpyro
+        import numpyro.distributions as dist
+        from numpyro.infer import MCMC, NUTS
+
+        def model():
+            _ = numpyro.sample(
+                "gamma", dist.ZeroSumNormal(1, event_shape=(10,)), infer={"event_dims": ["groups"]}
+            )
+
+        mcmc = MCMC(NUTS(model), num_warmup=10, num_samples=10)
+        mcmc.run(PRNGKey(0))
+        inference_data = from_numpyro(mcmc, coords={"groups": np.arange(10)})
+        assert inference_data.posterior.gamma.dims == ("chain", "draw", "groups")
+        assert "groups" in inference_data.posterior.gamma.coords
+
+    @pytest.mark.xfail
+    def test_mcmc_inferred_dims_univariate(self):
+        import numpyro
+        import numpyro.distributions as dist
+        from numpyro.infer import MCMC, NUTS
+        import jax.numpy as jnp
+
+        def model():
+            alpha = numpyro.sample("alpha", dist.Normal(0, 1))
+            sigma = numpyro.sample("sigma", dist.HalfNormal(1))
+            with numpyro.plate("obs_idx", 3):
+                # mu is plated by obs_idx, but isnt broadcasted to the plate shape
+                # the expected behavior is that this should cause a failure
+                mu = numpyro.deterministic("mu", alpha)
+                return numpyro.sample("y", dist.Normal(mu, sigma), obs=jnp.array([-1, 0, 1]))
+
+        mcmc = MCMC(NUTS(model), num_warmup=10, num_samples=10)
+        mcmc.run(PRNGKey(0))
+        inference_data = from_numpyro(mcmc, coords={"obs_idx": np.arange(3)})
+        assert inference_data.posterior.mu.dims == ("chain", "draw", "obs_idx")
+        assert "obs_idx" in inference_data.posterior.mu.coords
+
+    def test_mcmc_extra_event_dims(self):
+        import numpyro
+        import numpyro.distributions as dist
+        from numpyro.infer import MCMC, NUTS
+
+        def model():
+            gamma = numpyro.sample("gamma", dist.ZeroSumNormal(1, event_shape=(10,)))
+            _ = numpyro.deterministic("gamma_plus1", gamma + 1)
+
+        mcmc = MCMC(NUTS(model), num_warmup=10, num_samples=10)
+        mcmc.run(PRNGKey(0))
+        inference_data = from_numpyro(
+            mcmc, coords={"groups": np.arange(10)}, extra_event_dims={"gamma_plus1": ["groups"]}
+        )
+        assert inference_data.posterior.gamma_plus1.dims == ("chain", "draw", "groups")
+        assert "groups" in inference_data.posterior.gamma_plus1.coords
+
+    def test_mcmc_predictions_infer_dims(
+        self, data, eight_schools_params, predictions_data, predictions_params
+    ):
+        inference_data = self.get_inference_data(
+            data, eight_schools_params, predictions_data, predictions_params, infer_dims=True
+        )
+        assert inference_data.predictions.obs.dims == ("chain", "draw", "J")
+        assert "J" in inference_data.predictions.obs.coords

arviz/wrappers/base.py

@@ -197,7 +197,7 @@ class SamplingWrapper:
         """Check that all methods listed are implemented.
 
         Not all functions that require refitting need to have all the methods implemented in
-        order to work properly. This function shoulg be used before using the SamplingWrapper and
+        order to work properly. This function should be used before using the SamplingWrapper and
         its subclasses to get informative error messages.
 
         Parameters

arviz/wrappers/wrap_stan.py

@@ -44,7 +44,7 @@ class StanSamplingWrapper(SamplingWrapper):
         excluded_observed_data : str
             Variable name containing the pointwise log likelihood data of the excluded
             data. As PyStan cannot call C++ functions and log_likelihood__i is already
-            calculated *during* the simultion, instead of the value on which to evaluate
+            calculated *during* the simulation, instead of the value on which to evaluate
             the likelihood, ``log_likelihood__i`` expects a string so it can extract the
             corresponding data from the InferenceData object.
         """

{arviz-0.21.0.dist-info → arviz-0.22.0.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: arviz
-Version: 0.21.0
+Version: 0.22.0
 Summary: Exploratory analysis of Bayesian models
 Home-page: http://github.com/arviz-devs/arviz
 Author: ArviZ Developers
@@ -22,12 +22,12 @@ Requires-Python: >=3.10
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: setuptools>=60.0.0
-Requires-Dist: matplotlib>=3.5
-Requires-Dist: numpy>=1.23.0
-Requires-Dist: scipy>=1.9.0
+Requires-Dist: matplotlib>=3.8
+Requires-Dist: numpy>=1.26.0
+Requires-Dist: scipy>=1.11.0
 Requires-Dist: packaging
-Requires-Dist: pandas>=1.5.0
-Requires-Dist: xarray>=2022.6.0
+Requires-Dist: pandas>=2.1.0
+Requires-Dist: xarray>=2023.7.0
 Requires-Dist: h5netcdf>=1.0.2
 Requires-Dist: typing-extensions>=4.1.0
 Requires-Dist: xarray-einstats>=0.3
@@ -39,7 +39,7 @@ Requires-Dist: contourpy; extra == "all"
 Requires-Dist: ujson; extra == "all"
 Requires-Dist: dask[distributed]; extra == "all"
 Requires-Dist: zarr<3,>=2.5.0; extra == "all"
-Requires-Dist: xarray-datatree; extra == "all"
+Requires-Dist: xarray>=2024.11.0; extra == "all"
 Requires-Dist: dm-tree>=0.1.8; extra == "all"
 Provides-Extra: preview
 Requires-Dist: arviz-base[h5netcdf]; extra == "preview"

{arviz-0.21.0.dist-info → arviz-0.22.0.dist-info}/RECORD

@@ -1,4 +1,4 @@
-arviz/__init__.py,sha256=1PWUe3oeouP_C8JmAXhMrHarplymbqLyWqm950C-aAI,10397
+arviz/__init__.py,sha256=IOdU7uIJbHurf--mKztuR4Yq-fVHSLyU_neL4nK89KE,10590
 arviz/labels.py,sha256=w4-t0qdJzjKrqRyhzbtk6ucoMIAxle1HpHYlH7up06Q,6828
 arviz/preview.py,sha256=Fmff8j9Zlvgi5w2PRwnbkEOioJY8fK9p1SWQWjTl4N8,1314
 arviz/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
@@ -9,16 +9,16 @@ arviz/data/__init__.py,sha256=SG2umdZ8uVNYKVBoVYPy5tNxJnzLdyw0spPMjMTae5k,1558
 arviz/data/base.py,sha256=PALdVidyCxJqje_za4XPwXH010qAMJt4VzzeOflUCC8,21365
 arviz/data/converters.py,sha256=4QU5_0OU84HV3gSXXY_N5bd6S9ft81lfpnCT-8E8qyc,7933
 arviz/data/datasets.py,sha256=wPi23KZI-w4HrhjmY1LUg0Rj0aJobSJ4WO3LBrjfZQc,5392
-arviz/data/inference_data.py,sha256=GYmQzz26c614IX_XVvmoTA0ycaC2pZs3B48hd2HYhrA,93619
+arviz/data/inference_data.py,sha256=BvZsTROxRIGc6c4Jg_NQrjnlfm_WquPyhJi88KBEkis,94545
 arviz/data/io_beanmachine.py,sha256=QQVBD6rftvs6_kLIb4Vm1QzQ6BsS0J9DTrzw2Jj4ob8,3745
 arviz/data/io_cmdstan.py,sha256=8YX9Nfkx4LqjfKms4s4GTOkOjZNelb6SukvRZRHY6iM,38994
 arviz/data/io_cmdstanpy.py,sha256=iSr8ciKBFoIa1tJGHEX-2JKkUJRyaTXzRXf-5mu8q5U,42991
-arviz/data/io_datatree.py,sha256=Ag_7LHAejqt0Zxbu350u8WBY44nGsFY7eyhKa0Wi7FE,502
+arviz/data/io_datatree.py,sha256=QNj6Fkbv9HiJbCMbQmwGNPkwhYD4BkeMN10syhV9-J0,498
 arviz/data/io_dict.py,sha256=HM4ke-NuopsPnIdU-UGMtppAnj3vcbkBti8eROpMRTs,17698
 arviz/data/io_emcee.py,sha256=zsJJqwlyXLN_TfI6hgKz97p4N30NYTVvQSrIIpiLmB0,11844
 arviz/data/io_json.py,sha256=lrSP_9abfUW_5E8TwnG4hsS5HNHzAHZQCJTynl_tXKY,1256
 arviz/data/io_netcdf.py,sha256=cCxVnXSCTzWP3SU7cM4SqBiRRK9txFOsm-MchzNUzM4,2336
-arviz/data/io_numpyro.py,sha256=_U6owmsY5hrQU7C5NPw43s76Swmy_YqXw4UZYfZwzU4,14241
+arviz/data/io_numpyro.py,sha256=FgIoHp62eSgTOwNLIdfYhlOWIuPey55pvz949zwyX2I,18567
 arviz/data/io_pyjags.py,sha256=AG2ckAuygSxbjULVFQCJQSM72GnUoTMi3T94aQRJYKQ,13271
 arviz/data/io_pyro.py,sha256=JYywUGUU1Qil_ahLuDYhYFafQAKB-y1kIipXdfH_vnQ,12740
 arviz/data/io_pystan.py,sha256=nRTU6yujilQCKERxzN7LIVwZplfvFNb-Y9Jk9YVJQLk,41700
@@ -30,7 +30,7 @@ arviz/data/example_data/code/radon/radon.json,sha256=XwpiyGRrqkBP02zWz00s6z-d00V
 arviz/data/example_data/data/centered_eight.nc,sha256=jvw6uv4MeW65rqe2lJDU4kAKM8V1BO9JMuHHEFhJF28,654694
 arviz/data/example_data/data/non_centered_eight.nc,sha256=r7kyd10HyJTTRQs4OlSCXPVt3T-nLsPd3g-bcPYnPmA,836647
 arviz/plots/__init__.py,sha256=atWhfLXHAD6zaTjbdWdTaJrPTNBl3XpcSCY4etgw_cY,1514
-arviz/plots/autocorrplot.py,sha256=f-rYtnZtxgOns5sIT6pdw-EdU74lAVmAeV9b6rKLsgw,5666
+arviz/plots/autocorrplot.py,sha256=veeZNEhHoDBzR-mGNm-JOP1gBoSpilpy2E6lOgcoWKk,5926
 arviz/plots/bfplot.py,sha256=TKCkk60dgIk70CNWz9pHDXb1HGwHT_aLfNgFH6jDz9c,4367
 arviz/plots/bpvplot.py,sha256=Pf0ME21_teSQd_7CcuYE2MxfDJHcy5eW7Ms7yIDZSCA,12505
 arviz/plots/compareplot.py,sha256=Z8usSMEeQKs4GmkziDR4dVzSh3Ocd4ySfiNDZVaFOUc,6078
@@ -44,12 +44,12 @@ arviz/plots/energyplot.py,sha256=znEDPYpWaTIX0XpdVoyhXOITJ4A8BYkI9t1TVhJq4Qo,479
 arviz/plots/essplot.py,sha256=ch0DjUQDILk4ohpSUR-9VHejGFb6Xmelly-qa-fKb9E,11741
 arviz/plots/forestplot.py,sha256=37Wj4wFGjydZS4SpdqZ9qLxhZBo4rFxV_MWQnZAS7DA,11897
 arviz/plots/hdiplot.py,sha256=Pii9ZsuejEM-I24dn39muUL-yYKTfe2RWzAuU0W-3SI,7798
-arviz/plots/kdeplot.py,sha256=55mQPyIrvPKeJbjTsqWTExcff6QPvXFXG47XtDfPs_Y,11924
+arviz/plots/kdeplot.py,sha256=t-SJt3LIL1nThAsVM5npXZhRxqkGoCsfF1F0Fkj8ZV8,11924
 arviz/plots/khatplot.py,sha256=u03gmBG1xwxG07ASLZEJB-GsRhRHtUCtbglpC1N0aIg,8086
-arviz/plots/lmplot.py,sha256=LxR7RXkaAi5J8076isebVrtdk6UwbcTRekEymM9S6cY,11726
+arviz/plots/lmplot.py,sha256=ZKX0RNaUpQO4qYYDqRqc_yktNsfdUXjs4EGFU2Wem2o,12943
 arviz/plots/loopitplot.py,sha256=bFUO_Fy4z6u6E39GdaF4rIvc-COWNwF4A0n2kcmZBfA,8321
 arviz/plots/mcseplot.py,sha256=rsiz4E9M9p58YetAaF75gbenGIj4M0hapWnh9UJOXzY,6829
-arviz/plots/pairplot.py,sha256=yVxyLC7ms0SKpTunifacUyOblH8BW9CsQqE2Hv0ARP4,10407
+arviz/plots/pairplot.py,sha256=v-NCJIG6UG9cGIdFUWzW5S7Y29Ag5zE9zucxNSv46ME,10787
 arviz/plots/parallelplot.py,sha256=ZBEsHvnlmSXLRpSwP-KUwzgWBC2S4siKXFGJnLf7uAE,7125
 arviz/plots/plot_utils.py,sha256=VyVR50HrZegdkWa6ZxtRnC_WJstooYvaB-xsDHf6kaQ,18337
 arviz/plots/posteriorplot.py,sha256=pC-5SQZOGq1F4opM_sQLxn4ZG2we4Y9ULV8yhxjGVdo,10952
@@ -74,13 +74,13 @@ arviz/plots/backends/bokeh/elpdplot.py,sha256=5bn_rH1Aixm0--BArAP4m4kuZMjWxR8ox-
 arviz/plots/backends/bokeh/energyplot.py,sha256=FKPYRaWwM32Vw0AHE0MIWBn9wZQeoYdP_YpYZGijIq0,4571
 arviz/plots/backends/bokeh/essplot.py,sha256=0i2E1TvlzzP9mTavlGlJ5hPoFW5tGWCkAPAKrVHbgYQ,5560
 arviz/plots/backends/bokeh/forestplot.py,sha256=TAaXMqF2F1G7BIeDdWas14C6pCKZ3RTNhLIjUm-_IQs,27466
-arviz/plots/backends/bokeh/hdiplot.py,sha256=bAhTPi9D7cw2ytSvjSIRD3g-PqpB9OFwusrQAnmKCYY,1538
+arviz/plots/backends/bokeh/hdiplot.py,sha256=O4sDsYDe94VsDRM0BFGblwQGKXE0WhA8Y8TmwX85vd8,1670
 arviz/plots/backends/bokeh/kdeplot.py,sha256=nKEgJfnP7NK2Y2cipF_RglEC6GpowaagH5wZP3iZq-U,9329
 arviz/plots/backends/bokeh/khatplot.py,sha256=Iz2C6YaQOhnucAFbjSWTha2HNLFmctXc4tt_rLdS6Ko,4663
-arviz/plots/backends/bokeh/lmplot.py,sha256=0VIu1CPngKFQZayOqASAYHfXU8s_SRzysjdmxkELRJo,5600
+arviz/plots/backends/bokeh/lmplot.py,sha256=BP1-5O-PqpkheKHe1vDaZTzbAmFP_YKs9K02c3kYWtA,6532
 arviz/plots/backends/bokeh/loopitplot.py,sha256=FWjcsSWGJNy4wM63_N2qpg5oECb3Cq-uLy3xDS5x6j4,7172
 arviz/plots/backends/bokeh/mcseplot.py,sha256=QHyXeANcVSeLRdPJGh5RihZmaYq8cAPjSAn6FGJXQRk,5960
-arviz/plots/backends/bokeh/pairplot.py,sha256=kNvB3pVcpWXz7GMjBKc0pNyJdbx7VYgyEQOZeFbVgKo,13416
+arviz/plots/backends/bokeh/pairplot.py,sha256=RI480xTSrCvMNd8YQx5sjcDDkIytkSuGeCfWYpYIm8o,12482
 arviz/plots/backends/bokeh/parallelplot.py,sha256=SNChOLWvcKxXuuJsIfWs9CNj7qDuVb95UZyp4CP1BQE,2230
 arviz/plots/backends/bokeh/posteriorplot.py,sha256=yBAlGo3lQpFBVro724eZtKnOHgVkY7LXupa-8_VtE7c,9380
 arviz/plots/backends/bokeh/ppcplot.py,sha256=fDeXXn1WtjjSs-285nDo76NKJTp2m5hIZ7V9S6iqA30,13307
@@ -104,11 +104,11 @@ arviz/plots/backends/matplotlib/essplot.py,sha256=C5RH8MNIOGpCPKHUsGfZ03aQ3CaXI8
 arviz/plots/backends/matplotlib/forestplot.py,sha256=TWmJqmoH8Rn2bc-g3Tx1QTb_RIfNUKN5PxPFRnF3flA,23228
 arviz/plots/backends/matplotlib/hdiplot.py,sha256=7KawWKFahxILp0dxodkhRni9oJMCkulLhLhRD9qoH60,1521
 arviz/plots/backends/matplotlib/kdeplot.py,sha256=wsTBABG3MLkMAoAzu22otJdXCVzUnUu3McIt9_Hjldc,5323
-arviz/plots/backends/matplotlib/khatplot.py,sha256=JajJXlDXJce1DvB0Eh7ueB4N6ZkqKHgLaGhck8EsaXs,7437
-arviz/plots/backends/matplotlib/lmplot.py,sha256=EeXiqQhfQBaKAhmzRMLLV5gNGELo_QPZ24MkGnDWUfM,5288
+arviz/plots/backends/matplotlib/khatplot.py,sha256=SUV-Bbb88mnvFwxzFUf2j66JMjVtJNHoqYTRE7DmC8Y,7735
+arviz/plots/backends/matplotlib/lmplot.py,sha256=CHFhvUZaexnkS0ZFOJWTB3t0fdCF915RbmR-L2-cuM4,5564
 arviz/plots/backends/matplotlib/loopitplot.py,sha256=glK-BP4NftmdZEK5sB7kM8SzoGWsDnBdDVDzV-fDdCg,4632
 arviz/plots/backends/matplotlib/mcseplot.py,sha256=kGuRHRnyQKZPxoiHp8S30RzK7qbL83AaFGbn7BNVOZo,5810
-arviz/plots/backends/matplotlib/pairplot.py,sha256=82MfXs8Gxt40pGqsj48hvyM0Xvc44Eskj2fQC-0AasM,13793
+arviz/plots/backends/matplotlib/pairplot.py,sha256=NTBMIyvOMmCHwlrHeriEkwInA1R7beA9M28AlqCw8dU,13595
 arviz/plots/backends/matplotlib/parallelplot.py,sha256=zxtO6CNsK_HSl7E2sH40x8OYoO9a5bPNJ6VPJTuDQbk,1450
 arviz/plots/backends/matplotlib/posteriorplot.py,sha256=dUJfGYWYv5Lzlbz9Tr5d9virVfJb7JsnGiSYU7CdmsI,10092
 arviz/plots/backends/matplotlib/ppcplot.py,sha256=3kPTVEUsGpMyr_P5OKgfAu_NHZNmdJWXvCmfhIlNieE,16134
@@ -137,27 +137,27 @@ arviz/plots/styles/arviz-whitegrid.mplstyle,sha256=IMjjlfG3wg7heUjcVrkez1SNoiMI6
 arviz/static/css/style.css,sha256=wcC7rvCT4E6TycEiw7YqxwyaaZ4tTRDGqMkYYjAqrao,5910
 arviz/static/html/icons-svg-inline.html,sha256=t-ChbtS1Gv8uZxc31DCJS8SuXDsLGUHoKgwv8zu6j2M,1343
 arviz/stats/__init__.py,sha256=kvrANzMkqyHMTdry7N5w836E2OP0tJM6bm5-G8OZaA0,721
-arviz/stats/density_utils.py,sha256=WCEkXCQsaycY_usA6xkorhIqr3_5ru7wgwE9eJJgdH8,32216
+arviz/stats/density_utils.py,sha256=wmPFJzEZR7KgKxwQb5pGhY-w-rnFZpMIavrhpt_6u9w,32215
 arviz/stats/diagnostics.py,sha256=COTy2c5ROAirCAK_UNo7kQnggN71maBRPwy54ZdabKE,32656
 arviz/stats/ecdf_utils.py,sha256=Wy38wL-bsHDlpyU9qnDjedYBvbP_6ZrzJuWo6NzD2Xg,11835
-arviz/stats/stats.py,sha256=0cpdhmTwEVNY6VedeYhl2D3zlXj2hn2Mp2yqVkufP94,89884
+arviz/stats/stats.py,sha256=Q4MMxSJPyElFzBXpCqA21bMb1jQfnmLAHVTPmDHGiSo,90129
 arviz/stats/stats_refitting.py,sha256=trbPC7LCnsb-n5D6g7J0bzXJCPfcDukJDniB4ud1z9E,5415
 arviz/stats/stats_utils.py,sha256=XG8ILPVs8Jbh_v7jzLfwMkm2HraT2j2-Hxe_kEYlLjQ,20076
 arviz/tests/__init__.py,sha256=TiS6C1IzwAXmNa8u36Y2xzL1CTTZm2PwzAtmZgoqepE,18
 arviz/tests/conftest.py,sha256=6U9WpKmYf38EVRoFZNBpV0CunQvESBFJG2SJ8IBEkL4,1270
 arviz/tests/helpers.py,sha256=qhsOhLtfyz-dC2yuT6ug0frYZlbims06BljJuEVDP6E,23593
 arviz/tests/base_tests/__init__.py,sha256=zg7V5_0DZrCz7ozETqET6bUpAvUUmmkSoLpJLwaIj2E,23
-arviz/tests/base_tests/test_data.py,sha256=L3vfscLm8aMVjd3MswJqmJ_J7Y2sNjwjCeJQb4TX9_Q,64296
+arviz/tests/base_tests/test_data.py,sha256=vnaYftgX_r58ra-Eo7ivAhUC8cSWAhf-zOLuDlWdZXA,64131
 arviz/tests/base_tests/test_data_zarr.py,sha256=sPWnIQ7vPhN5Ql3Dq4JENuSwQV5IeignQjy9BAYe1_A,5441
 arviz/tests/base_tests/test_diagnostics.py,sha256=pbuy1-nvTKWSHv0nnhXOhpG4e2uy-4GGZb4lxAdoMpw,20353
 arviz/tests/base_tests/test_diagnostics_numba.py,sha256=2G5O-7Hz66DSaHIZtjs2XL45RezYnXQZH6Dg2Ow-p4Q,2847
 arviz/tests/base_tests/test_helpers.py,sha256=PogHpWCMBEtkuzKt9jGQ8uIPg0cLDwztXxOnPSPNyVE,669
 arviz/tests/base_tests/test_labels.py,sha256=X08vTMmcgXkYGbE2Qnu_UUDSTAIvSNKdnyqoWwmj008,1686
 arviz/tests/base_tests/test_plot_utils.py,sha256=lwDZYDNrlEOKP-asJv6qu3sH_4y-FiHcFlqnMTpZyhw,11771
-arviz/tests/base_tests/test_plots_bokeh.py,sha256=1JqUqLKUb1g4c4w41K3j_LCT4eqb3u1qGnYbUuJMHPE,39148
-arviz/tests/base_tests/test_plots_matplotlib.py,sha256=sRn-jWzBeCkk49i8DO_a1l9WXoia269sANOQfPpqYjQ,66197
+arviz/tests/base_tests/test_plots_bokeh.py,sha256=FDw3dp-M89EVsIAWvl7M17GXWyatRalYQJJHsbT5BzQ,41052
+arviz/tests/base_tests/test_plots_matplotlib.py,sha256=v_GwjrmWV18AyG-1aetUzpx2QqD9nqIrxsfVOKHtGY0,68533
 arviz/tests/base_tests/test_rcparams.py,sha256=b9ueOXd9C0xiUIqgS0qnzvalHFgTFK7sUqL8UAzgJNs,10851
-arviz/tests/base_tests/test_stats.py,sha256=zEK8uKvXBFpnLQPxoHkM5L9-KWqHQXvUcWAzKGMGgFQ,32809
+arviz/tests/base_tests/test_stats.py,sha256=QHVa8sSzr5FX8X0D8tGFntCz9aW032gLu-EOuC6RHA4,34322
 arviz/tests/base_tests/test_stats_ecdf_utils.py,sha256=p1FnQzlC0fjjKDFfhbHIrrbwAVhLiygH4J0aarx89A0,6038
 arviz/tests/base_tests/test_stats_numba.py,sha256=wGXgNuSO_gwJajoYtXSgpIe88PcBRyIkRihxC8paR-o,1582
 arviz/tests/base_tests/test_stats_utils.py,sha256=Udkw8tODs8mLt3_hO3HgNczrU0n09IJrML2agXF-upQ,13864
@@ -168,16 +168,16 @@ arviz/tests/external_tests/test_data_beanmachine.py,sha256=nwOJNJLrk5rY4M5YW-LT6
 arviz/tests/external_tests/test_data_cmdstan.py,sha256=jHy-dZrY4M7F4uYWf71fOxVwfPxgRpM9E3JAvpk03qA,16829
 arviz/tests/external_tests/test_data_cmdstanpy.py,sha256=uCSOJVowKXccCPLpAwCiihghx_WxnUVyR8r801Xhw_0,18753
 arviz/tests/external_tests/test_data_emcee.py,sha256=w-tsP74-n688C9-v_KIf0YxZg7S1WrhOdJUvaHS9e6I,6270
-arviz/tests/external_tests/test_data_numpyro.py,sha256=TB5IkxlU3pMsjIgnxyhQaCkRkPWaK_d_YHEcs7XaWNo,11366
+arviz/tests/external_tests/test_data_numpyro.py,sha256=cdUzv0MKPCmsfNLbvz7IgBxapGSJjG42bgaR_MfDmPg,16758
 arviz/tests/external_tests/test_data_pyjags.py,sha256=kqZfV8QRnAngO9obnAq5lKPIuJdVJ82sbkIfSr2tpqY,4547
 arviz/tests/external_tests/test_data_pyro.py,sha256=EaD_hZGALaSKQKK4OFgmuJ_1SsIYKessHQ7Jl9AKbw0,10771
 arviz/tests/external_tests/test_data_pystan.py,sha256=ebg_JXkmAhXRllP0otjyourGF_fUaKMkwRfrQO6Glwk,11792
 arviz/wrappers/__init__.py,sha256=d8GTUuBW_30LyDyk6qn2MAnvg-GZCeUw_i5SUPqaa1w,354
-arviz/wrappers/base.py,sha256=Vvh330pdzIvBEaikHsDP1ej6L2jCZZ0Dqj5TvUbYesI,9134
+arviz/wrappers/base.py,sha256=FNgPvd_tLCB5C2tRx1ngYjr4F5tEUuNrrLkStuyRXsE,9134
 arviz/wrappers/wrap_pymc.py,sha256=ltKv55aG0WTWXVDJuff5TXkgJJ_ESLvlT-JPlh3yHAg,1143
-arviz/wrappers/wrap_stan.py,sha256=c40brlajoPcc3xk00xI9Hqc-y0xcbAmFAIZOtfXWeqo,5525
-arviz-0.21.0.dist-info/LICENSE,sha256=xllut76FgcGL5zbIRvuRc7aezPbvlMUTWJPsVr2Sugg,11358
-arviz-0.21.0.dist-info/METADATA,sha256=3zSEB4npy44RXZtTxQVEkIvjvCqbDXQ9s6vsAyxvXmg,8848
-arviz-0.21.0.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
-arviz-0.21.0.dist-info/top_level.txt,sha256=5MFvqrTtYRWsIx-SjKuFIUHtrnVJq0Ngd0Nc2_etQhE,6
-arviz-0.21.0.dist-info/RECORD,,
+arviz/wrappers/wrap_stan.py,sha256=sIy38fXg4Ln_0CM6xONDwOJg1Y6FwNM_JQErv3a-8_c,5526
+arviz-0.22.0.dist-info/LICENSE,sha256=xllut76FgcGL5zbIRvuRc7aezPbvlMUTWJPsVr2Sugg,11358
+arviz-0.22.0.dist-info/METADATA,sha256=co8xturD_y3pE5nwukUw29h3QPDd4fyVFXylpFidKvY,8851
+arviz-0.22.0.dist-info/WHEEL,sha256=tZoeGjtWxWRfdplE7E3d45VPlLNQnvbKiYnx7gwAy8A,92
+arviz-0.22.0.dist-info/top_level.txt,sha256=5MFvqrTtYRWsIx-SjKuFIUHtrnVJq0Ngd0Nc2_etQhE,6
+arviz-0.22.0.dist-info/RECORD,,