e2bc076a6676458ffa7c0b45f17c0290e763808a,arviz/stats/diagnostics.py,,ess,#Any#,119

Before Change

        "identity": _ess_identity,
    }

    methods_relative = {
        "bulk": _ress_bulk,
        "tail": _ress_tail,
        "quantile": _ress_quantile,
        "mean": _ress_mean,
        "sd": _ress_sd,
        "median": _ress_median,
        "mad": _ress_mad,
        "z_scale": _ress_z_scale,
        "folded": _ress_folded,
        "split": _ress_split,
        "identity": _ress_identity,
    }

    if method not in methods:
        raise TypeError(
            "ESS method {} not found. Valid methods are:\n{}".format(method, "\n    ".join(methods))

After Change

    if isinstance(data, np.ndarray):
        data = np.atleast_2d(data)
        if len(data.shape) < 3:
            if prob is not None:
                return ess_func(  // pylint: disable=unexpected-keyword-arg
                    data, prob=prob, relative=relative
                )
            else:
                return ess_func(data, relative=relative)
        else:
            msg = (
                "Only uni-dimensional ndarray variables are supported."
                " Please transform first to dataset with `az.convert_to_dataset`."
            )
            raise TypeError(msg)

    dataset = convert_to_dataset(data, group="posterior")
    var_names = _var_names(var_names, dataset)

    dataset = dataset if var_names is None else dataset[var_names]

Instances