[Kwant] Re: extracting scattering states momenta with Kwant