Wrong shape in block covariance matrix

[fedezuc]

asrpy/asrpy/asr_utils.py

Line 504 in 14956f4

U = np.zeros([len(np.arange(0, n_times - 1, window)), n_ch**2])

When data has shape of (4,17302), window = 100 gives error

ValueError: cannot reshape array of size 2768 into shape (174,16)

And it does the same with every n_samples in the shape of

xxxx02

I think it is an approximation problem when creating the matrix U.
I patched it by creating the matrix U during the first iteration, like this

    for k in range(0, window):
        idx_range = np.minimum(n_times - 1,
                               np.arange(k, n_times + k - 2, window))
        if k == 0:
             U = np.zeros([data[idx_range].shape[0], n_ch**2])

        U = U + np.reshape(data[idx_range].reshape([-1, 1, n_ch])*data[idx_range].reshape(-1, n_ch, 1), U.shape)

[DWonGH]

I encountered this problem, and the proposed fix seemed to work.