arviz 0.22.0__tar.gz → 0.23.0__tar.gz

{arviz-0.22.0 → arviz-0.23.0}/CHANGELOG.md

@@ -1,5 +1,13 @@
 # Change Log
 
+## v0.23.0 (2025 Des 9)
+
+### Maintenance and fixes
+- Fix numpyro jax incompatibility. ([2465](https://github.com/arviz-devs/arviz/pull/2465))
+- Avoid closing unloaded files in `from_netcdf()` ([2463](https://github.com/arviz-devs/arviz/issues/2463))
+- Fix sign error in lp parsed in from_numpyro ([2468](https://github.com/arviz-devs/arviz/issues/2468))
+- Fix attrs persistance in idata-datatree conversions ([2476](https://github.com/arviz-devs/arviz/issues/2476))
+
 ## v0.22.0 (2025 Jul 9)
 
 ### New features

{arviz-0.22.0 → arviz-0.23.0}/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: arviz
-Version: 0.22.0
+Version: 0.23.0
 Summary: Exploratory analysis of Bayesian models
 Home-page: http://github.com/arviz-devs/arviz
 Author: ArviZ Developers
@@ -20,9 +20,42 @@ Classifier: Topic :: Scientific/Engineering :: Visualization
 Classifier: Topic :: Scientific/Engineering :: Mathematics
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: setuptools>=60.0.0
+Requires-Dist: matplotlib>=3.8
+Requires-Dist: numpy>=1.26.0
+Requires-Dist: scipy>=1.11.0
+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: xarray-einstats>=0.3
 Provides-Extra: all
+Requires-Dist: numba; extra == "all"
+Requires-Dist: netcdf4; extra == "all"
+Requires-Dist: bokeh>=3; extra == "all"
+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>=2024.11.0; extra == "all"
+Requires-Dist: dm-tree>=0.1.8; extra == "all"
 Provides-Extra: preview
-License-File: LICENSE
+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 → arviz-0.23.0}/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.0"
 
 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-0.22.0 → arviz-0.23.0}/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-0.22.0 → arviz-0.23.0}/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-0.22.0 → arviz-0.23.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-0.22.0 → arviz-0.23.0}/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-0.22.0 → arviz-0.23.0}/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-0.22.0 → arviz-0.23.0}/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-0.22.0 → arviz-0.23.0}/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-0.22.0 → arviz-0.23.0}/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-0.23.0/arviz/preview.py

@@ -0,0 +1,58 @@
+# pylint: disable=unused-import,unused-wildcard-import,wildcard-import,invalid-name
+"""Expose features from arviz-xyz refactored packages inside ``arviz.preview`` namespace."""
+import logging
+
+_log = logging.getLogger(__name__)
+
+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)
+except ModuleNotFoundError:
+    _status = "arviz_base not installed"
+    _log.info(_status)
+except ImportError:
+    _status = "Unable to import arviz_base"
+    _log.info(_status, exc_info=True)
+
+info += _status + "\n"
+
+try:
+    from arviz_stats import *
+
+    # 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)
+except ImportError:
+    _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)
+except ModuleNotFoundError:
+    _status = "arviz_plots not installed"
+    _log.info(_status)
+except ImportError:
+    _status = "Unable to import arviz_plots"
+    _log.info(_status, exc_info=True)
+
+info += _status + "\n"
+
+# clean namespace
+del logging, _status, _log

{arviz-0.22.0 → arviz-0.23.0}/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-0.22.0 → arviz-0.23.0}/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-0.22.0 → arviz-0.23.0}/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-0.22.0 → arviz-0.23.0}/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-0.22.0 → arviz-0.23.0}/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-0.22.0 → arviz-0.23.0}/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 → arviz-0.23.0}/arviz.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: arviz
-Version: 0.22.0
+Version: 0.23.0
 Summary: Exploratory analysis of Bayesian models
 Home-page: http://github.com/arviz-devs/arviz
 Author: ArviZ Developers
@@ -20,9 +20,42 @@ Classifier: Topic :: Scientific/Engineering :: Visualization
 Classifier: Topic :: Scientific/Engineering :: Mathematics
 Requires-Python: >=3.10
 Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: setuptools>=60.0.0
+Requires-Dist: matplotlib>=3.8
+Requires-Dist: numpy>=1.26.0
+Requires-Dist: scipy>=1.11.0
+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: xarray-einstats>=0.3
 Provides-Extra: all
+Requires-Dist: numba; extra == "all"
+Requires-Dist: netcdf4; extra == "all"
+Requires-Dist: bokeh>=3; extra == "all"
+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>=2024.11.0; extra == "all"
+Requires-Dist: dm-tree>=0.1.8; extra == "all"
 Provides-Extra: preview
-License-File: LICENSE
+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/arviz/preview.py

@@ -1,48 +0,0 @@
-# pylint: disable=unused-import,unused-wildcard-import,wildcard-import,invalid-name
-"""Expose features from arviz-xyz refactored packages inside ``arviz.preview`` namespace."""
-import logging
-
-_log = logging.getLogger(__name__)
-
-info = ""
-
-try:
-    from arviz_base import *
-
-    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)
-except ImportError:
-    status = "Unable to import arviz_base"
-    _log.info(status, exc_info=True)
-
-info += status + "\n"
-
-try:
-    from arviz_stats import *
-
-    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)
-except ImportError:
-    status = "Unable to import arviz_stats"
-    _log.info(status, exc_info=True)
-info += status + "\n"
-
-try:
-    from arviz_plots import *
-
-    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)
-except ImportError:
-    status = "Unable to import arviz_plots"
-    _log.info(status, exc_info=True)
-
-info += status + "\n"