wavelet scattering transform python