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

arviz/__init__.py

@@ -1,9 +1,10 @@
 # pylint: disable=wildcard-import,invalid-name,wrong-import-position
 """ArviZ is a library for exploratory analysis of Bayesian models."""
-__version__ = "0.22.0"
+__version__ = "0.23.1"
 
 import logging
 import os
+import re
 
 from matplotlib.colors import LinearSegmentedColormap
 from matplotlib.pyplot import style
@@ -11,6 +12,45 @@ import matplotlib as mpl
 from packaging import version
 
 
+def _warn_once_per_day():
+    from .preview import info
+
+    # skip warning if all 3 arviz subpackages are already installed
+    pat = re.compile(r"arviz_(base|stats|plots) available")
+    if len(pat.findall(info)) == 3:
+        return
+
+    import datetime
+    from warnings import warn
+    from pathlib import Path
+
+    warning_dir = Path.home() / "arviz_data"
+    warning_dir.mkdir(exist_ok=True)
+
+    stamp_file = warning_dir / "daily_warning"
+    today = datetime.date.today()
+
+    if stamp_file.exists():
+        last_date = datetime.date.fromisoformat(stamp_file.read_text().strip())
+    else:
+        last_date = None
+
+    if last_date != today:
+        warn(
+            "\nArviZ is undergoing a major refactor to improve flexibility and extensibility "
+            "while maintaining a user-friendly interface."
+            "\nSome upcoming changes may be backward incompatible."
+            "\nFor details and migration guidance, visit: "
+            "https://python.arviz.org/en/latest/user_guide/migration_guide.html",
+            FutureWarning,
+        )
+
+        stamp_file.write_text(today.isoformat())
+
+
+_warn_once_per_day()
+
+
 class Logger(logging.Logger):
     """Override Logger to avoid repeated messages."""
 
@@ -333,4 +373,4 @@ except ModuleNotFoundError:
 
 
 # clean namespace
-del os, logging, LinearSegmentedColormap, Logger, mpl
+del os, re, logging, version, LinearSegmentedColormap, Logger, mpl

arviz/data/converters.py

@@ -2,6 +2,7 @@
 
 import numpy as np
 import xarray as xr
+import pandas as pd
 
 try:
     from tree import is_nested
@@ -44,6 +45,8 @@ def convert_to_inference_data(obj, *, group="posterior", coords=None, dims=None,
             | dict: creates an xarray dataset as the only group
             | numpy array: creates an xarray dataset as the only group, gives the
                          array an arbitrary name
+            | object with __array__: converts to numpy array, then creates an xarray dataset as
+                         the only group, gives the array an arbitrary name
     group : str
         If `obj` is a dict or numpy array, assigns the resulting xarray
         dataset to this group. Default: "posterior".
@@ -115,6 +118,13 @@ def convert_to_inference_data(obj, *, group="posterior", coords=None, dims=None,
         dataset = dict_to_dataset(obj, coords=coords, dims=dims)
     elif isinstance(obj, np.ndarray):
         dataset = dict_to_dataset({"x": obj}, coords=coords, dims=dims)
+    elif (
+        hasattr(obj, "__array__")
+        and callable(getattr(obj, "__array__"))
+        and (not isinstance(obj, pd.DataFrame))
+    ):
+        obj = obj.__array__()
+        dataset = dict_to_dataset({"x": obj}, coords=coords, dims=dims)
     elif isinstance(obj, (list, tuple)) and isinstance(obj[0], str) and obj[0].endswith(".csv"):
         if group == "sample_stats":
             kwargs["posterior"] = kwargs.pop(group)
@@ -129,6 +139,7 @@ def convert_to_inference_data(obj, *, group="posterior", coords=None, dims=None,
             "pytree (if 'dm-tree' is installed)",
             "netcdf filename",
             "numpy array",
+            "object with __array__",
             "pystan fit",
             "emcee fit",
             "pyro mcmc fit",

arviz/data/inference_data.py

@@ -430,11 +430,12 @@ class InferenceData(Mapping[str, xr.Dataset]):
                         if re.search(key, group):
                             group_kws = kws
                 group_kws.setdefault("engine", engine)
-                with xr.open_dataset(filename, group=f"{base_group}/{group}", **group_kws) as data:
-                    if rcParams["data.load"] == "eager":
+                data = xr.open_dataset(filename, group=f"{base_group}/{group}", **group_kws)
+                if rcParams["data.load"] == "eager":
+                    with data:
                         groups[group] = data.load()
-                    else:
-                        groups[group] = data
+                else:
+                    groups[group] = data
 
             with xr.open_dataset(filename, engine=engine, group=base_group) as data:
                 attrs.update(data.load().attrs)
@@ -540,7 +541,9 @@ class InferenceData(Mapping[str, xr.Dataset]):
                 "xarray must be have DataTree in order to use InferenceData.to_datatree. "
                 "Update to xarray>=2024.11.0"
             ) from err
-        return DataTree.from_dict({group: ds for group, ds in self.items()})
+        dt = DataTree.from_dict({group: ds for group, ds in self.items()})
+        dt.attrs = self.attrs
+        return dt
 
     @staticmethod
     def from_datatree(datatree):
@@ -551,7 +554,8 @@ class InferenceData(Mapping[str, xr.Dataset]):
         datatree : DataTree
         """
         return InferenceData(
-            **{group: child.to_dataset() for group, child in datatree.children.items()}
+            attrs=datatree.attrs,
+            **{group: child.to_dataset() for group, child in datatree.children.items()},
         )
 
     def to_dict(self, groups=None, filter_groups=None):
@@ -811,7 +815,7 @@ class InferenceData(Mapping[str, xr.Dataset]):
         if version.parse(zarr.__version__) >= version.parse("3.0.0.dev0"):
             raise ImportError(
                 "Found zarr>=3, which is not supported by ArviZ. Instead, you can use "
-                "'dt = InfereceData.to_datatree' followed by 'dt.to_zarr()' "
+                "'dt = InferenceData.to_datatree' followed by 'dt.to_zarr()' "
                 "(needs xarray>=2024.11.0)"
             )
 

arviz/data/io_numpyro.py

@@ -241,7 +241,10 @@ class NumPyroConverter:
                 continue
             name = rename_key.get(stat, stat)
             value = value.copy()
-            data[name] = value
+            if stat == "potential_energy":
+                data[name] = -value
+            else:
+                data[name] = value
             if stat == "num_steps":
                 data["tree_depth"] = np.log2(value).astype(int) + 1
         return dict_to_dataset(

arviz/data/io_pyjags.py

@@ -277,7 +277,7 @@ def _extract_arviz_dict_from_inference_data(
 
 
 def _convert_arviz_dict_to_pyjags_dict(
-    samples: tp.Mapping[str, np.ndarray]
+    samples: tp.Mapping[str, np.ndarray],
 ) -> tp.Mapping[str, np.ndarray]:
     """
     Convert and ArviZ dictionary to a PyJAGS dictionary.

arviz/plots/bpvplot.py

@@ -251,7 +251,7 @@ def plot_bpv(
     total_pp_samples = predictive_dataset.sizes["chain"] * predictive_dataset.sizes["draw"]
 
     for key in coords.keys():
-        coords[key] = np.where(np.in1d(observed[key], coords[key]))[0]
+        coords[key] = np.where(np.isin(observed[key], coords[key]))[0]
 
     obs_plotters = filter_plotters_list(
         list(

arviz/plots/dotplot.py

@@ -2,6 +2,7 @@
 
 import numpy as np
 
+
 from ..rcparams import rcParams
 from .plot_utils import get_plotting_function
 
@@ -148,6 +149,7 @@ def plot_dot(
         raise ValueError("marker argument is valid only for matplotlib backend")
 
     values = np.ravel(values)
+    values = values[np.isfinite(values)]
     values.sort()
 
     if hdi_prob is None:

arviz/plots/forestplot.py

@@ -51,7 +51,7 @@ def plot_forest(
     data : InferenceData
         Any object that can be converted to an :class:`arviz.InferenceData` object
         Refer to documentation of :func:`arviz.convert_to_dataset` for details.
-    kind : {"foresplot", "ridgeplot"}, default "forestplot"
+    kind : {"forestplot", "ridgeplot"}, default "forestplot"
         Specify the kind of plot:
 
         * The ``kind="forestplot"`` generates credible intervals, where the central points are the
@@ -75,8 +75,8 @@ def plot_forest(
         interpret `var_names` as substrings of the real variables names. If "regex",
         interpret `var_names` as regular expressions on the real variables names. See
         :ref:`this section <common_filter_vars>` for usage examples.
-    transform : callable, optional
-        Function to transform data (defaults to None i.e.the identity function).
+    transform : callable or dict, optional
+        Function to transform the data. Defaults to None, i.e., the identity function.
     coords : dict, optional
         Coordinates of ``var_names`` to be plotted. Passed to :meth:`xarray.Dataset.sel`.
         See :ref:`this section <common_coords>` for usage examples.
@@ -228,7 +228,19 @@ def plot_forest(
 
     datasets = [convert_to_dataset(datum) for datum in reversed(data)]
     if transform is not None:
-        datasets = [transform(dataset) for dataset in datasets]
+        if callable(transform):
+            datasets = [transform(dataset) for dataset in datasets]
+        elif isinstance(transform, dict):
+            transformed_datasets = []
+            for dataset in datasets:
+                new_dataset = dataset.copy()
+                for var_name, func in transform.items():
+                    if var_name in new_dataset:
+                        new_dataset[var_name] = func(new_dataset[var_name])
+                transformed_datasets.append(new_dataset)
+            datasets = transformed_datasets
+        else:
+            raise ValueError("transform must be either a callable or a dict {var_name: callable}")
     datasets = get_coords(
         datasets, list(reversed(coords)) if isinstance(coords, (list, tuple)) else coords
     )

arviz/plots/ppcplot.py

@@ -304,7 +304,7 @@ def plot_ppc(
     pp_sample_ix = np.random.choice(total_pp_samples, size=num_pp_samples, replace=False)
 
     for key in coords.keys():
-        coords[key] = np.where(np.in1d(observed_data[key], coords[key]))[0]
+        coords[key] = np.where(np.isin(observed_data[key], coords[key]))[0]
 
     obs_plotters = filter_plotters_list(
         list(

arviz/preview.py

@@ -8,41 +8,51 @@ info = ""
 
 try:
     from arviz_base import *
+    import arviz_base as base
 
-    status = "arviz_base available, exposing its functions as part of arviz.preview"
-    _log.info(status)
+    _status = "arviz_base available, exposing its functions as part of arviz.preview"
+    _log.info(_status)
 except ModuleNotFoundError:
-    status = "arviz_base not installed"
-    _log.info(status)
+    _status = "arviz_base not installed"
+    _log.info(_status)
 except ImportError:
-    status = "Unable to import arviz_base"
-    _log.info(status, exc_info=True)
+    _status = "Unable to import arviz_base"
+    _log.info(_status, exc_info=True)
 
-info += status + "\n"
+info += _status + "\n"
 
 try:
     from arviz_stats import *
 
-    status = "arviz_stats available, exposing its functions as part of arviz.preview"
-    _log.info(status)
+    # the base computational module fron arviz_stats will override the alias to arviz-base
+    # arviz.stats.base will still be available
+    import arviz_base as base
+    import arviz_stats as stats
+
+    _status = "arviz_stats available, exposing its functions as part of arviz.preview"
+    _log.info(_status)
 except ModuleNotFoundError:
-    status = "arviz_stats not installed"
-    _log.info(status)
+    _status = "arviz_stats not installed"
+    _log.info(_status)
 except ImportError:
-    status = "Unable to import arviz_stats"
-    _log.info(status, exc_info=True)
-info += status + "\n"
+    _status = "Unable to import arviz_stats"
+    _log.info(_status, exc_info=True)
+info += _status + "\n"
 
 try:
     from arviz_plots import *
+    import arviz_plots as plots
 
-    status = "arviz_plots available, exposing its functions as part of arviz.preview"
-    _log.info(status)
+    _status = "arviz_plots available, exposing its functions as part of arviz.preview"
+    _log.info(_status)
 except ModuleNotFoundError:
-    status = "arviz_plots not installed"
-    _log.info(status)
+    _status = "arviz_plots not installed"
+    _log.info(_status)
 except ImportError:
-    status = "Unable to import arviz_plots"
-    _log.info(status, exc_info=True)
+    _status = "Unable to import arviz_plots"
+    _log.info(_status, exc_info=True)
+
+info += _status + "\n"
 
-info += status + "\n"
+# clean namespace
+del logging, _status, _log

arviz/rcparams.py

@@ -12,11 +12,11 @@ from pathlib import Path
 from typing import Any, Dict
 from typing_extensions import Literal
 
-NO_GET_ARGS: bool = False
+NO_GET_ARGS: bool = False  # pylint: disable=invalid-name
 try:
     from typing_extensions import get_args
 except ImportError:
-    NO_GET_ARGS = True
+    NO_GET_ARGS = True  # pylint: disable=invalid-name
 
 
 import numpy as np

arviz/stats/stats.py

@@ -1,7 +1,6 @@
 # pylint: disable=too-many-lines
 """Statistical functions in ArviZ."""
 
-import itertools
 import warnings
 from copy import deepcopy
 from typing import List, Optional, Tuple, Union, Mapping, cast, Callable
@@ -11,14 +10,14 @@ import pandas as pd
 import scipy.stats as st
 from xarray_einstats import stats
 import xarray as xr
-from scipy.optimize import minimize
+from scipy.optimize import minimize, LinearConstraint, Bounds
 from typing_extensions import Literal
 
-NO_GET_ARGS: bool = False
+NO_GET_ARGS: bool = False  # pylint: disable=invalid-name
 try:
     from typing_extensions import get_args
 except ImportError:
-    NO_GET_ARGS = True
+    NO_GET_ARGS = True  # pylint: disable=invalid-name
 
 from .. import _log
 from ..data import InferenceData, convert_to_dataset, convert_to_inference_data, extract
@@ -225,37 +224,23 @@ def compare(
     if method.lower() == "stacking":
         rows, cols, ic_i_val = _ic_matrix(ics, ic_i)
         exp_ic_i = np.exp(ic_i_val / scale_value)
-        km1 = cols - 1
-
-        def w_fuller(weights):
-            return np.concatenate((weights, [max(1.0 - np.sum(weights), 0.0)]))
 
         def log_score(weights):
-            w_full = w_fuller(weights)
-            score = 0.0
-            for i in range(rows):
-                score += np.log(np.dot(exp_ic_i[i], w_full))
-            return -score
+            return -np.sum(np.log(exp_ic_i @ weights))
 
         def gradient(weights):
-            w_full = w_fuller(weights)
-            grad = np.zeros(km1)
-            for k, i in itertools.product(range(km1), range(rows)):
-                grad[k] += (exp_ic_i[i, k] - exp_ic_i[i, km1]) / np.dot(exp_ic_i[i], w_full)
-            return -grad
-
-        theta = np.full(km1, 1.0 / cols)
-        bounds = [(0.0, 1.0) for _ in range(km1)]
-        constraints = [
-            {"type": "ineq", "fun": lambda x: -np.sum(x) + 1.0},
-            {"type": "ineq", "fun": np.sum},
-        ]
+            denominator = exp_ic_i @ weights
+            return -np.sum(exp_ic_i / denominator[:, np.newaxis], axis=0)
+
+        theta = np.full(cols, 1.0 / cols)
+        bounds = Bounds(lb=np.zeros(cols), ub=np.ones(cols))
+        constraints = LinearConstraint(np.ones(cols), lb=1.0, ub=1.0)
 
-        weights = minimize(
+        minimize_result = minimize(
             fun=log_score, x0=theta, jac=gradient, bounds=bounds, constraints=constraints
         )
 
-        weights = w_fuller(weights["x"])
+        weights = minimize_result["x"]
         ses = ics["se"]
 
     elif method.lower() == "bb-pseudo-bma":

arviz/tests/base_tests/test_data.py

@@ -1510,6 +1510,15 @@ class TestDataTree:
             assert_identical(ds, idata_back[group])
         assert all(group in dt.children for group in idata.groups())
 
+    def test_datatree_attrs(self):
+        idata = load_arviz_data("centered_eight")
+        idata.attrs = {"not": "empty"}
+        assert idata.attrs
+        dt = idata.to_datatree()
+        idata_back = from_datatree(dt)
+        assert dt.attrs == idata.attrs
+        assert idata_back.attrs == idata.attrs
+
 
 class TestConversions:
     def test_id_conversion_idempotent(self):
@@ -1652,3 +1661,19 @@ class TestExtractDataset:
         post = extract(idata, num_samples=10)
         assert post.sizes["sample"] == 10
         assert post.attrs == idata.posterior.attrs
+
+
+def test_convert_to_inference_data_with_array_like():
+    class ArrayLike:
+        def __init__(self, data):
+            self._data = np.asarray(data)
+
+        def __array__(self):
+            return self._data
+
+    array_like = ArrayLike(np.random.randn(4, 100))
+    idata = convert_to_inference_data(array_like, group="posterior")
+
+    assert hasattr(idata, "posterior")
+    assert "x" in idata.posterior.data_vars
+    assert idata.posterior["x"].shape == (4, 100)

arviz/tests/base_tests/test_plots_matplotlib.py

@@ -2176,5 +2176,22 @@ def test_plot_autocorr_coords(coords, expected_vars):
     idata = load_arviz_data("centered_eight")
 
     axes = plot_autocorr(idata, var_names=expected_vars, coords=coords, show=False)
+    assert axes is not None
+
+
+def test_plot_forest_with_transform():
+    """Test if plot_forest runs successfully with a transform dictionary."""
+    data = xr.Dataset(
+        {
+            "var1": (["chain", "draw"], np.array([[1, 2, 3], [4, 5, 6]])),
+            "var2": (["chain", "draw"], np.array([[7, 8, 9], [10, 11, 12]])),
+        },
+        coords={"chain": [0, 1], "draw": [0, 1, 2]},
+    )
+    transform_dict = {
+        "var1": lambda x: x + 1,
+        "var2": lambda x: x * 2,
+    }
 
+    axes = plot_forest(data, transform=transform_dict, show=False)
     assert axes is not None

arviz/tests/base_tests/test_stats_ecdf_utils.py

@@ -13,9 +13,9 @@ from ...stats.ecdf_utils import (
 try:
     import numba  # pylint: disable=unused-import
 
-    numba_options = [True, False]
+    numba_options = [True, False]  # pylint: disable=invalid-name
 except ImportError:
-    numba_options = [False]
+    numba_options = [False]  # pylint: disable=invalid-name
 
 
 def test_compute_ecdf():

arviz/tests/external_tests/test_data_numpyro.py

@@ -1,4 +1,4 @@
-# pylint: disable=no-member, invalid-name, redefined-outer-name
+# pylint: disable=no-member, invalid-name, redefined-outer-name, too-many-public-methods
 from collections import namedtuple
 import numpy as np
 import pytest
@@ -409,3 +409,26 @@ class TestDataNumPyro:
         )
         assert inference_data.predictions.obs.dims == ("chain", "draw", "J")
         assert "J" in inference_data.predictions.obs.coords
+
+    def test_potential_energy_sign_conversion(self):
+        """Test that potential energy is converted to log probability (lp) with correct sign."""
+        import numpyro
+        import numpyro.distributions as dist
+        from numpyro.infer import MCMC, NUTS
+
+        num_samples = 10
+
+        def simple_model():
+            numpyro.sample("x", dist.Normal(0, 1))
+
+        nuts_kernel = NUTS(simple_model)
+        mcmc = MCMC(nuts_kernel, num_samples=num_samples, num_warmup=5)
+        mcmc.run(PRNGKey(0), extra_fields=["potential_energy"])
+
+        # Get the raw extra fields from NumPyro
+        extra_fields = mcmc.get_extra_fields(group_by_chain=True)
+        # Convert to ArviZ InferenceData
+        inference_data = from_numpyro(mcmc)
+        arviz_lp = inference_data["sample_stats"]["lp"].values
+
+        np.testing.assert_array_equal(arviz_lp, -extra_fields["potential_energy"])

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

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: arviz
-Version: 0.22.0
+Version: 0.23.1
 Summary: Exploratory analysis of Bayesian models
 Home-page: http://github.com/arviz-devs/arviz
 Author: ArviZ Developers
@@ -29,7 +29,8 @@ Requires-Dist: packaging
 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: h5py
+Requires-Dist: typing_extensions>=4.1.0
 Requires-Dist: xarray-einstats>=0.3
 Provides-Extra: all
 Requires-Dist: numba; extra == "all"
@@ -45,6 +46,17 @@ Provides-Extra: preview
 Requires-Dist: arviz-base[h5netcdf]; extra == "preview"
 Requires-Dist: arviz-stats[xarray]; extra == "preview"
 Requires-Dist: arviz-plots; extra == "preview"
+Dynamic: author
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: license
+Dynamic: license-file
+Dynamic: provides-extra
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
 
 <img src="https://raw.githubusercontent.com/arviz-devs/arviz-project/main/arviz_logos/ArviZ.png#gh-light-mode-only" width=200></img>
 <img src="https://raw.githubusercontent.com/arviz-devs/arviz-project/main/arviz_logos/ArviZ_white.png#gh-dark-mode-only" width=200></img>

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

@@ -1,15 +1,15 @@
-arviz/__init__.py,sha256=IOdU7uIJbHurf--mKztuR4Yq-fVHSLyU_neL4nK89KE,10590
+arviz/__init__.py,sha256=HBMDwUgE8CTGybvpN9HEjphGDofJi3ywCFFVbzt3KPc,11756
 arviz/labels.py,sha256=w4-t0qdJzjKrqRyhzbtk6ucoMIAxle1HpHYlH7up06Q,6828
-arviz/preview.py,sha256=Fmff8j9Zlvgi5w2PRwnbkEOioJY8fK9p1SWQWjTl4N8,1314
+arviz/preview.py,sha256=GuAwiStRYSooI4mG9j7JFpr3YxBfg_93DlgroORQBUg,1644
 arviz/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-arviz/rcparams.py,sha256=lQnBprbDihEcsP5FujoJetGna4T8nc20JnuVf1Wn-1k,21031
+arviz/rcparams.py,sha256=PRQYTnvwgGZrQkb1li3n-ZPdBaowDqfW51VTUa4dOjU,21095
 arviz/sel_utils.py,sha256=xvAYENhGXDTrhaT4Itlk1SJQUUGZ6BGcR04fPFgvzdM,6951
 arviz/utils.py,sha256=-q3eAeficZcLakOt6UXN0eyVx6uIVVpkqA_ky8FRSjE,26699
 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/converters.py,sha256=TeiKpYCk4BSqkBzlSmMwAl3PNYvKaEy5sUMpQlYvt88,8411
 arviz/data/datasets.py,sha256=wPi23KZI-w4HrhjmY1LUg0Rj0aJobSJ4WO3LBrjfZQc,5392
-arviz/data/inference_data.py,sha256=BvZsTROxRIGc6c4Jg_NQrjnlfm_WquPyhJi88KBEkis,94545
+arviz/data/inference_data.py,sha256=vfueSyWiqYlVOvn_Y3xdYrNvCuVYaP3QX0x74emH9xI,94639
 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
@@ -18,8 +18,8 @@ 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=FgIoHp62eSgTOwNLIdfYhlOWIuPey55pvz949zwyX2I,18567
-arviz/data/io_pyjags.py,sha256=AG2ckAuygSxbjULVFQCJQSM72GnUoTMi3T94aQRJYKQ,13271
+arviz/data/io_numpyro.py,sha256=zJiL-U6x4hxHYs3mzTXdzF4iT2t4BBM5vh3zfU3I6eE,18668
+arviz/data/io_pyjags.py,sha256=PqljKXfXN13vxVJsLiNOdz_IGDhT-sTd-6nhn70LKVg,13272
 arviz/data/io_pyro.py,sha256=JYywUGUU1Qil_ahLuDYhYFafQAKB-y1kIipXdfH_vnQ,12740
 arviz/data/io_pystan.py,sha256=nRTU6yujilQCKERxzN7LIVwZplfvFNb-Y9Jk9YVJQLk,41700
 arviz/data/io_zarr.py,sha256=PeSBz-zHDzmwJq3sWzxASnjrfbd-hULJsl8FjK46YQQ,1163
@@ -32,17 +32,17 @@ arviz/data/example_data/data/non_centered_eight.nc,sha256=r7kyd10HyJTTRQs4OlSCXP
 arviz/plots/__init__.py,sha256=atWhfLXHAD6zaTjbdWdTaJrPTNBl3XpcSCY4etgw_cY,1514
 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/bpvplot.py,sha256=JcZPn-ON6nCZYiWTxaxGRMkkTkAEZHE4FE3cX2oYa-c,12505
 arviz/plots/compareplot.py,sha256=Z8usSMEeQKs4GmkziDR4dVzSh3Ocd4ySfiNDZVaFOUc,6078
 arviz/plots/densityplot.py,sha256=6477ZljpBCcZRw0SUwcTO4FYjxqw_qYsJupWNo-jCok,10895
 arviz/plots/distcomparisonplot.py,sha256=gVNQUN0VX7hC527fcUk1oxtQRdIl5mrltU95c0Nra9k,7184
 arviz/plots/distplot.py,sha256=xWXOsN-pPBwhHrEjC6lbIJdn-17DtpMueSnj6YzWlX4,8472
-arviz/plots/dotplot.py,sha256=9HTMeT1ZuZ4Vauxvg4TjsvvNnwORG8WWO2HistJwHiU,7736
+arviz/plots/dotplot.py,sha256=jpXzHhlBxUuyJF6hnAUpaiuyL1zwLOJgq4CYwmqEtWs,7778
 arviz/plots/ecdfplot.py,sha256=eYasPwOYEmzqx82d6SyDg_iPyXkyFuOKnbOBjPrndH0,13112
 arviz/plots/elpdplot.py,sha256=NKqPkwTj9BWDzwMnG0cPeLmYBGMX_meP9F6bqvTwLKY,6433
 arviz/plots/energyplot.py,sha256=znEDPYpWaTIX0XpdVoyhXOITJ4A8BYkI9t1TVhJq4Qo,4797
 arviz/plots/essplot.py,sha256=ch0DjUQDILk4ohpSUR-9VHejGFb6Xmelly-qa-fKb9E,11741
-arviz/plots/forestplot.py,sha256=37Wj4wFGjydZS4SpdqZ9qLxhZBo4rFxV_MWQnZAS7DA,11897
+arviz/plots/forestplot.py,sha256=Cp8Xfk9CI_ZNOQj2qTMDy5UWkE2wVabOLForDuEcWxY,12505
 arviz/plots/hdiplot.py,sha256=Pii9ZsuejEM-I24dn39muUL-yYKTfe2RWzAuU0W-3SI,7798
 arviz/plots/kdeplot.py,sha256=t-SJt3LIL1nThAsVM5npXZhRxqkGoCsfF1F0Fkj8ZV8,11924
 arviz/plots/khatplot.py,sha256=u03gmBG1xwxG07ASLZEJB-GsRhRHtUCtbglpC1N0aIg,8086
@@ -53,7 +53,7 @@ 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
-arviz/plots/ppcplot.py,sha256=UPTtXDWHf3wFAb-apNPGcz8qw9CQwINGml_2YkYI-iM,13967
+arviz/plots/ppcplot.py,sha256=QwcgZTuUDijuXK9g3AB8Lc8ShK2URSL8Wc4Jeu5IxZM,13967
 arviz/plots/rankplot.py,sha256=lz0swHs6EBe-gXn4udP1Um3RS-EatsOAmguYqGMlIjU,8648
 arviz/plots/separationplot.py,sha256=Fx_QVeFUcF45fm7nn06pt0qubOzvH8QMU1cw5RLyaik,5491
 arviz/plots/traceplot.py,sha256=dwcF7rsjMAIxZ_LPv7Z8os01uQZHXTkDFWEBtsbzI9k,10216
@@ -140,14 +140,14 @@ arviz/stats/__init__.py,sha256=kvrANzMkqyHMTdry7N5w836E2OP0tJM6bm5-G8OZaA0,721
 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=Q4MMxSJPyElFzBXpCqA21bMb1jQfnmLAHVTPmDHGiSo,90129
+arviz/stats/stats.py,sha256=vowy6JVX1BkKynlrcBtDGVZ2ws3wWeiiTrMBFC0E6iY,89743
 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=vnaYftgX_r58ra-Eo7ivAhUC8cSWAhf-zOLuDlWdZXA,64131
+arviz/tests/base_tests/test_data.py,sha256=TZ9638lELh_rFqp0RnS5WThELGoqiPd39fSBXgcxB8Y,64910
 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
@@ -155,10 +155,10 @@ arviz/tests/base_tests/test_helpers.py,sha256=PogHpWCMBEtkuzKt9jGQ8uIPg0cLDwztXx
 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=FDw3dp-M89EVsIAWvl7M17GXWyatRalYQJJHsbT5BzQ,41052
-arviz/tests/base_tests/test_plots_matplotlib.py,sha256=v_GwjrmWV18AyG-1aetUzpx2QqD9nqIrxsfVOKHtGY0,68533
+arviz/tests/base_tests/test_plots_matplotlib.py,sha256=1MpAZHAlG9hc_vRTfXBy7-qxLLsZQNfns1qAVo6J9X8,69097
 arviz/tests/base_tests/test_rcparams.py,sha256=b9ueOXd9C0xiUIqgS0qnzvalHFgTFK7sUqL8UAzgJNs,10851
 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_ecdf_utils.py,sha256=1T_9jYyuCWMEwTbps4AFHRDzEF8THcdFSeQN2oZIwVw,6102
 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
 arviz/tests/base_tests/test_utils.py,sha256=Auggtvwv3Y9STS8Tbram-IQe5IhewkwFN14CTcjRd5M,12533
@@ -168,7 +168,7 @@ 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=cdUzv0MKPCmsfNLbvz7IgBxapGSJjG42bgaR_MfDmPg,16758
+arviz/tests/external_tests/test_data_numpyro.py,sha256=ehI-xkxoub25xPIU1GyAhJDs4BpVwV_DLS0HYiL0Odw,17678
 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
@@ -176,8 +176,8 @@ arviz/wrappers/__init__.py,sha256=d8GTUuBW_30LyDyk6qn2MAnvg-GZCeUw_i5SUPqaa1w,35
 arviz/wrappers/base.py,sha256=FNgPvd_tLCB5C2tRx1ngYjr4F5tEUuNrrLkStuyRXsE,9134
 arviz/wrappers/wrap_pymc.py,sha256=ltKv55aG0WTWXVDJuff5TXkgJJ_ESLvlT-JPlh3yHAg,1143
 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,,
+arviz-0.23.1.dist-info/licenses/LICENSE,sha256=xllut76FgcGL5zbIRvuRc7aezPbvlMUTWJPsVr2Sugg,11358
+arviz-0.23.1.dist-info/METADATA,sha256=PYF49qwd_wLFdlbmfyNVwiDg-AzZk3GPxEey5YHi6io,9109
+arviz-0.23.1.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+arviz-0.23.1.dist-info/top_level.txt,sha256=5MFvqrTtYRWsIx-SjKuFIUHtrnVJq0Ngd0Nc2_etQhE,6
+arviz-0.23.1.dist-info/RECORD,,

{arviz-0.22.0.dist-info → arviz-0.23.1.dist-info}/WHEEL

@@ -1,5 +1,5 @@
 Wheel-Version: 1.0
-Generator: bdist_wheel (0.45.1)
+Generator: setuptools (80.9.0)
 Root-Is-Purelib: true
 Tag: py3-none-any