e7baa93f4f397c2955cb30c83bf483f26e67f2f8,arviz/stats/diagnostics.py,,_rhat_ufunc,#Any#,229

Before Change

    `xr.apply_ufunc(_neff_ufunc, ..., input_core_dims=(("chain", "draw"),))
    
    target = np.empty(ary.shape[:-2])
    for idxs in itertools.product(*[np.arange(d) for d in target.shape]):
        idxs = list(idxs)
        idxs.append(Ellipsis)
        target[tuple(idxs)] = _get_rhat(ary[tuple(idxs)])
    return target


def geweke(values, first=.1, last=.5, intervals=20):

After Change

    
    target = np.empty(ary.shape[:-2])
    for idx in np.ndindex(target.shape):
        target[idx] = _get_rhat(ary[idx])
    return target

Instances