e7baa93f4f397c2955cb30c83bf483f26e67f2f8,arviz/stats/diagnostics.py,,_neff_ufunc,#Any#,107

Before Change

    for idxs in itertools.product(*[np.arange(d) for d in target.shape]):
        idxs = list(idxs)
        idxs.append(Ellipsis)
        target[tuple(idxs)] = _get_neff(ary[tuple(idxs)])
    return target


def autocorr(x):

After Change

    `xr.apply_ufunc(_neff_ufunc, ..., input_core_dims=(("chain", "draw"),))
    
    target = np.empty(ary.shape[:-2])
    for idx in np.ndindex(target.shape):
        target[idx] = _get_neff(ary[idx])
    return target


def _get_neff(sample_array):

Instances