Could you change this to `mapped from (-pi, pi) to (0, 2 * pi)`? I had to do a bit of digging to figure out what you were doing here, where it would have been obvious if the docstring had said this.