08be819a5fd8edd74a59dcc363c9e7694ca009d2,neurodsp/utils/outliers.py,,remove_nans,#,8

Before Change

    

    sig_nans = np.isnan(sig)
    sig_removed = sig[np.where(~np.isnan(sig))]

    return sig_removed, sig_nans

After Change

    sig_nans = np.isnan(sig)

    if sig.ndim == 1:
        sig_removed = sig[np.where(~sig_nans)]
    elif sig.ndim == 2:
        sig_removed = sig[~sig_nans].reshape(sig_nans.shape[0], sum(~sig_nans[0, :]))
        sig_nans = sig_nans[0, :]
    else:
        raise ValueError("Only 1d or 2d arrays supported.")

    return sig_removed, sig_nans


def restore_nans(sig, sig_nans, dtype=float):

Instances