7d11df30d7b62e47321775c3cd41fbe51a0305b2,pyemma/_base/subset.py,,globalise,#Any#Any#Any#Any#Any#,48

Before Change

def globalise(data, axis, active_set, default_value, n_centers):
    shape_org = _np.shape(data)
    ndim = data.ndim
    n = n_centers if ndim == 1 else n_centers * shape_org[1]
    array = _np.asarray([default_value]).repeat(n)
    if ndim == 1:
        array[active_set] = data
    elif ndim == 2:
        array = array.reshape(-1, shape_org[1])
        if axis == 0:
            array[active_set, :] = data
        elif axis == 1:
            array[:, active_set] = data

After Change

    elif data.ndim == 2:
        expanded_shape = list(data.shape)
        expanded_shape[axis] = n_centers
        array = _np.asarray([default_value]).repeat(
            expanded_shape[0] * expanded_shape[1]).reshape(expanded_shape)
        if axis == 0:
            array[active_set, :] = data
        elif axis == 1:
            array[:, active_set] = data

Instances