arviz 0.23.3__py3-none-any.whl → 1.0.0rc0__py3-none-any.whl

arviz/sel_utils.py

@@ -1,223 +0,0 @@
-"""Utilities for selecting and iterating on xarray objects."""
-
-from itertools import product, tee
-
-import numpy as np
-import xarray as xr
-
-from .labels import BaseLabeller
-
-__all__ = ["xarray_sel_iter", "xarray_var_iter", "xarray_to_ndarray"]
-
-
-def selection_to_string(selection):
-    """Convert dictionary of coordinates to a string for labels.
-
-    Parameters
-    ----------
-    selection : dict[Any] -> Any
-
-    Returns
-    -------
-    str
-        key1: value1, key2: value2, ...
-    """
-    return ", ".join([f"{v}" for _, v in selection.items()])
-
-
-def make_label(var_name, selection, position="below"):
-    """Consistent labelling for plots.
-
-    Parameters
-    ----------
-    var_name : str
-       Name of the variable
-
-    selection : dict[Any] -> Any
-        Coordinates of the variable
-    position : str
-        Whether to position the coordinates' label "below" (default) or "beside"
-        the name of the variable
-
-    Returns
-    -------
-    label
-        A text representation of the label
-    """
-    if selection:
-        sel = selection_to_string(selection)
-        if position == "below":
-            base = "{}\n{}"
-        elif position == "beside":
-            base = "{}[{}]"
-    else:
-        sel = ""
-        base = "{}{}"
-    return base.format(var_name, sel)
-
-
-def _dims(data, var_name, skip_dims):
-    return [dim for dim in data[var_name].dims if dim not in skip_dims]
-
-
-def _zip_dims(new_dims, vals):
-    return [dict(zip(new_dims, prod)) for prod in product(*vals)]
-
-
-def xarray_sel_iter(data, var_names=None, combined=False, skip_dims=None, reverse_selections=False):
-    """Convert xarray data to an iterator over variable names and selections.
-
-    Iterates over each var_name and all of its coordinates, returning the variable
-    names and selections that allow properly obtain the data from ``data`` as desired.
-
-    Parameters
-    ----------
-    data : xarray.Dataset
-        Posterior data in an xarray
-
-    var_names : iterator of strings (optional)
-        Should be a subset of data.data_vars. Defaults to all of them.
-
-    combined : bool
-        Whether to combine chains or leave them separate
-
-    skip_dims : set
-        dimensions to not iterate over
-
-    reverse_selections : bool
-        Whether to reverse selections before iterating.
-
-    Returns
-    -------
-    Iterator of (var_name: str, selection: dict(str, any))
-        The string is the variable name, the dictionary are coordinate names to values,.
-        To get the values of the variable at these coordinates, do
-        ``data[var_name].sel(**selection)``.
-    """
-    if skip_dims is None:
-        skip_dims = set()
-
-    if combined:
-        skip_dims = skip_dims.union({"chain", "draw"})
-    else:
-        skip_dims.add("draw")
-
-    if var_names is None:
-        if isinstance(data, xr.Dataset):
-            var_names = list(data.data_vars)
-        elif isinstance(data, xr.DataArray):
-            var_names = [data.name]
-            data = {data.name: data}
-
-    for var_name in var_names:
-        if var_name in data:
-            new_dims = _dims(data, var_name, skip_dims)
-            vals = [list(dict.fromkeys(data[var_name][dim].values)) for dim in new_dims]
-            dims = _zip_dims(new_dims, vals)
-            idims = _zip_dims(new_dims, [range(len(v)) for v in vals])
-            if reverse_selections:
-                dims = reversed(dims)
-                idims = reversed(idims)
-
-            for selection, iselection in zip(dims, idims):
-                yield var_name, selection, iselection
-
-
-def xarray_var_iter(
-    data, var_names=None, combined=False, skip_dims=None, reverse_selections=False, dim_order=None
-):
-    """Convert xarray data to an iterator over vectors.
-
-    Iterates over each var_name and all of its coordinates, returning the 1d
-    data.
-
-    Parameters
-    ----------
-    data : xarray.Dataset
-        Posterior data in an xarray
-
-    var_names : iterator of strings (optional)
-        Should be a subset of data.data_vars. Defaults to all of them.
-
-    combined : bool
-        Whether to combine chains or leave them separate
-
-    skip_dims : set
-        dimensions to not iterate over
-
-    reverse_selections : bool
-        Whether to reverse selections before iterating.
-
-    dim_order: list
-        Order for the first dimensions. Skips dimensions not found in the variable.
-
-    Returns
-    -------
-    Iterator of (str, dict(str, any), np.array)
-        The string is the variable name, the dictionary are coordinate names to values,
-        and the array are the values of the variable at those coordinates.
-    """
-    data_to_sel = data
-    if var_names is None and isinstance(data, xr.DataArray):
-        data_to_sel = {data.name: data}
-
-    if isinstance(dim_order, str):
-        dim_order = [dim_order]
-
-    for var_name, selection, iselection in xarray_sel_iter(
-        data,
-        var_names=var_names,
-        combined=combined,
-        skip_dims=skip_dims,
-        reverse_selections=reverse_selections,
-    ):
-        selected_data = data_to_sel[var_name].sel(**selection)
-        if dim_order is not None:
-            dim_order_selected = [dim for dim in dim_order if dim in selected_data.dims]
-            if dim_order_selected:
-                selected_data = selected_data.transpose(*dim_order_selected, ...)
-        yield var_name, selection, iselection, selected_data.values
-
-
-def xarray_to_ndarray(data, *, var_names=None, combined=True, label_fun=None):
-    """Take xarray data and unpacks into variables and data into list and numpy array respectively.
-
-    Assumes that chain and draw are in coordinates
-
-    Parameters
-    ----------
-    data: xarray.DataSet
-        Data in an xarray from an InferenceData object. Examples include posterior or sample_stats
-
-    var_names: iter
-        Should be a subset of data.data_vars not including chain and draws. Defaults to all of them
-
-    combined: bool
-        Whether to combine chain into one array
-
-    Returns
-    -------
-    var_names: list
-        List of variable names
-    data: np.array
-        Data values
-    """
-    if label_fun is None:
-        label_fun = BaseLabeller().make_label_vert
-    data_to_sel = data
-    if var_names is None and isinstance(data, xr.DataArray):
-        data_to_sel = {data.name: data}
-
-    iterator1, iterator2 = tee(xarray_sel_iter(data, var_names=var_names, combined=combined))
-    vars_and_sel = list(iterator1)
-    unpacked_var_names = [
-        label_fun(var_name, selection, isel) for var_name, selection, isel in vars_and_sel
-    ]
-
-    # Merge chains and variables, check dtype to be compatible with divergences data
-    data0 = data_to_sel[vars_and_sel[0][0]].sel(**vars_and_sel[0][1])
-    unpacked_data = np.empty((len(unpacked_var_names), data0.size), dtype=data0.dtype)
-    for idx, (var_name, selection, _) in enumerate(iterator2):
-        unpacked_data[idx] = data_to_sel[var_name].sel(**selection).values.flatten()
-
-    return unpacked_var_names, unpacked_data

arviz/static/css/style.css

@@ -1,340 +0,0 @@
-/* CSS stylesheet for displaying InferenceData objects in jupyterlab.
- *
- */
-
-:root {
-  --xr-font-color0: var(--jp-content-font-color0, rgba(0, 0, 0, 1));
-  --xr-font-color2: var(--jp-content-font-color2, rgba(0, 0, 0, 0.54));
-  --xr-font-color3: var(--jp-content-font-color3, rgba(0, 0, 0, 0.38));
-  --xr-border-color: var(--jp-border-color2, #e0e0e0);
-  --xr-disabled-color: var(--jp-layout-color3, #bdbdbd);
-  --xr-background-color: var(--jp-layout-color0, white);
-  --xr-background-color-row-even: var(--jp-layout-color1, white);
-  --xr-background-color-row-odd: var(--jp-layout-color2, #eeeeee);
-}
-
-html[theme=dark],
-body.vscode-dark {
-  --xr-font-color0: rgba(255, 255, 255, 1);
-  --xr-font-color2: rgba(255, 255, 255, 0.54);
-  --xr-font-color3: rgba(255, 255, 255, 0.38);
-  --xr-border-color: #1F1F1F;
-  --xr-disabled-color: #515151;
-  --xr-background-color: #111111;
-  --xr-background-color-row-even: #111111;
-  --xr-background-color-row-odd: #313131;
-}
-
-.xr-wrap {
-  display: block;
-  min-width: 300px;
-  max-width: 700px;
-}
-
-.xr-text-repr-fallback {
-  /* fallback to plain text repr when CSS is not injected (untrusted notebook) */
-  display: none;
-}
-
-.xr-header {
-  padding-top: 6px;
-  padding-bottom: 6px;
-  margin-bottom: 4px;
-  border-bottom: solid 1px var(--xr-border-color);
-}
-
-.xr-header > div,
-.xr-header > ul {
-  display: inline;
-  margin-top: 0;
-  margin-bottom: 0;
-}
-
-.xr-obj-type,
-.xr-array-name {
-  margin-left: 2px;
-  margin-right: 10px;
-}
-
-.xr-obj-type {
-  color: var(--xr-font-color2);
-}
-
-.xr-sections {
-  padding-left: 0 !important;
-  display: grid;
-  grid-template-columns: 150px auto auto 1fr 20px 20px;
-}
-
-.xr-sections.group-sections {
-  grid-template-columns: auto;
-}
-
-.xr-section-item {
-  display: contents;
-}
-
-.xr-section-item input {
-  display: none;
-}
-
-.xr-section-item input + label {
-  color: var(--xr-disabled-color);
-}
-
-.xr-section-item input:enabled + label {
-  cursor: pointer;
-  color: var(--xr-font-color2);
-}
-
-.xr-section-item input:enabled + label:hover {
-  color: var(--xr-font-color0);
-}
-
-.xr-section-summary {
-  grid-column: 1;
-  color: var(--xr-font-color2);
-  font-weight: 500;
-}
-
-.xr-section-summary > span {
-  display: inline-block;
-  padding-left: 0.5em;
-}
-
-.xr-section-summary-in:disabled + label {
-  color: var(--xr-font-color2);
-}
-
-.xr-section-summary-in + label:before {
-  display: inline-block;
-  content: '►';
-  font-size: 11px;
-  width: 15px;
-  text-align: center;
-}
-
-.xr-section-summary-in:disabled + label:before {
-  color: var(--xr-disabled-color);
-}
-
-.xr-section-summary-in:checked + label:before {
-  content: '▼';
-}
-
-.xr-section-summary-in:checked + label > span {
-  display: none;
-}
-
-.xr-section-summary,
-.xr-section-inline-details {
-  padding-top: 4px;
-  padding-bottom: 4px;
-}
-
-.xr-section-inline-details {
-  grid-column: 2 / -1;
-}
-
-.xr-section-details {
-  display: none;
-  grid-column: 1 / -1;
-  margin-bottom: 5px;
-}
-
-.xr-section-summary-in:checked ~ .xr-section-details {
-  display: contents;
-}
-
-.xr-array-wrap {
-  grid-column: 1 / -1;
-  display: grid;
-  grid-template-columns: 20px auto;
-}
-
-.xr-array-wrap > label {
-  grid-column: 1;
-  vertical-align: top;
-}
-
-.xr-preview {
-  color: var(--xr-font-color3);
-}
-
-.xr-array-preview,
-.xr-array-data {
-  padding: 0 5px !important;
-  grid-column: 2;
-}
-
-.xr-array-data,
-.xr-array-in:checked ~ .xr-array-preview {
-  display: none;
-}
-
-.xr-array-in:checked ~ .xr-array-data,
-.xr-array-preview {
-  display: inline-block;
-}
-
-.xr-dim-list {
-  display: inline-block !important;
-  list-style: none;
-  padding: 0 !important;
-  margin: 0;
-}
-
-.xr-dim-list li {
-  display: inline-block;
-  padding: 0;
-  margin: 0;
-}
-
-.xr-dim-list:before {
-  content: '(';
-}
-
-.xr-dim-list:after {
-  content: ')';
-}
-
-.xr-dim-list li:not(:last-child):after {
-  content: ',';
-  padding-right: 5px;
-}
-
-.xr-has-index {
-  font-weight: bold;
-}
-
-.xr-var-list,
-.xr-var-item {
-  display: contents;
-}
-
-.xr-var-item > div,
-.xr-var-item label,
-.xr-var-item > .xr-var-name span {
-  background-color: var(--xr-background-color-row-even);
-  margin-bottom: 0;
-}
-
-.xr-var-item > .xr-var-name:hover span {
-  padding-right: 5px;
-}
-
-.xr-var-list > li:nth-child(odd) > div,
-.xr-var-list > li:nth-child(odd) > label,
-.xr-var-list > li:nth-child(odd) > .xr-var-name span {
-  background-color: var(--xr-background-color-row-odd);
-}
-
-.xr-var-name {
-  grid-column: 1;
-}
-
-.xr-var-dims {
-  grid-column: 2;
-}
-
-.xr-var-dtype {
-  grid-column: 3;
-  text-align: right;
-  color: var(--xr-font-color2);
-}
-
-.xr-var-preview {
-  grid-column: 4;
-}
-
-.xr-var-name,
-.xr-var-dims,
-.xr-var-dtype,
-.xr-preview,
-.xr-attrs dt {
-  white-space: nowrap;
-  overflow: hidden;
-  text-overflow: ellipsis;
-  padding-right: 10px;
-}
-
-.xr-var-name:hover,
-.xr-var-dims:hover,
-.xr-var-dtype:hover,
-.xr-attrs dt:hover {
-  overflow: visible;
-  width: auto;
-  z-index: 1;
-}
-
-.xr-var-attrs,
-.xr-var-data {
-  display: none;
-  background-color: var(--xr-background-color) !important;
-  padding-bottom: 5px !important;
-}
-
-.xr-var-attrs-in:checked ~ .xr-var-attrs,
-.xr-var-data-in:checked ~ .xr-var-data {
-  display: block;
-}
-
-.xr-var-data > table {
-  float: right;
-}
-
-.xr-var-name span,
-.xr-var-data,
-.xr-attrs {
-  padding-left: 25px !important;
-}
-
-.xr-attrs,
-.xr-var-attrs,
-.xr-var-data {
-  grid-column: 1 / -1;
-}
-
-dl.xr-attrs {
-  padding: 0;
-  margin: 0;
-  display: grid;
-  grid-template-columns: 125px auto;
-}
-
-.xr-attrs dt,
-.xr-attrs dd {
-  padding: 0;
-  margin: 0;
-  float: left;
-  padding-right: 10px;
-  width: auto;
-}
-
-.xr-attrs dt {
-  font-weight: normal;
-  grid-column: 1;
-}
-
-.xr-attrs dt:hover span {
-  display: inline-block;
-  background: var(--xr-background-color);
-  padding-right: 10px;
-}
-
-.xr-attrs dd {
-  grid-column: 2;
-  white-space: pre-wrap;
-  word-break: break-all;
-}
-
-.xr-icon-database,
-.xr-icon-file-text2 {
-  display: inline-block;
-  vertical-align: middle;
-  width: 1em;
-  height: 1.5em !important;
-  stroke-width: 0;
-  stroke: currentColor;
-  fill: currentColor;
-}

arviz/static/html/icons-svg-inline.html

@@ -1,15 +0,0 @@
-<svg style="position: absolute; width: 0; height: 0; overflow: hidden">
-<defs>
-<symbol id="icon-database" viewBox="0 0 32 32">
-<path d="M16 0c-8.837 0-16 2.239-16 5v4c0 2.761 7.163 5 16 5s16-2.239 16-5v-4c0-2.761-7.163-5-16-5z"></path>
-<path d="M16 17c-8.837 0-16-2.239-16-5v6c0 2.761 7.163 5 16 5s16-2.239 16-5v-6c0 2.761-7.163 5-16 5z"></path>
-<path d="M16 26c-8.837 0-16-2.239-16-5v6c0 2.761 7.163 5 16 5s16-2.239 16-5v-6c0 2.761-7.163 5-16 5z"></path>
-</symbol>
-<symbol id="icon-file-text2" viewBox="0 0 32 32">
-<path d="M28.681 7.159c-0.694-0.947-1.662-2.053-2.724-3.116s-2.169-2.030-3.116-2.724c-1.612-1.182-2.393-1.319-2.841-1.319h-15.5c-1.378 0-2.5 1.121-2.5 2.5v27c0 1.378 1.122 2.5 2.5 2.5h23c1.378 0 2.5-1.122 2.5-2.5v-19.5c0-0.448-0.137-1.23-1.319-2.841zM24.543 5.457c0.959 0.959 1.712 1.825 2.268 2.543h-4.811v-4.811c0.718 0.556 1.584 1.309 2.543 2.268zM28 29.5c0 0.271-0.229 0.5-0.5 0.5h-23c-0.271 0-0.5-0.229-0.5-0.5v-27c0-0.271 0.229-0.5 0.5-0.5 0 0 15.499-0 15.5 0v7c0 0.552 0.448 1 1 1h7v19.5z"></path>
-<path d="M23 26h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z"></path>
-<path d="M23 22h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z"></path>
-<path d="M23 18h-14c-0.552 0-1-0.448-1-1s0.448-1 1-1h14c0.552 0 1 0.448 1 1s-0.448 1-1 1z"></path>
-</symbol>
-</defs>
-</svg>

arviz/stats/__init__.py

@@ -1,37 +0,0 @@
-# pylint: disable=wildcard-import
-"""Statistical tests and diagnostics for ArviZ."""
-from .density_utils import *
-from .diagnostics import *
-from .stats import *
-from .stats import _calculate_ics
-from .stats_refitting import *
-from .stats_utils import *
-
-__all__ = [
-    "apply_test_function",
-    "bayes_factor",
-    "bfmi",
-    "compare",
-    "hdi",
-    "kde",
-    "loo",
-    "loo_pit",
-    "psislw",
-    "r2_samples",
-    "r2_score",
-    "summary",
-    "waic",
-    "weight_predictions",
-    "ELPDData",
-    "ess",
-    "rhat",
-    "mcse",
-    "autocorr",
-    "autocov",
-    "make_ufunc",
-    "smooth_data",
-    "wrap_xarray_ufunc",
-    "reloo",
-    "_calculate_ics",
-    "psens",
-]