def _arctan2pi(b, a):
    ans = np.arctan2(b, a)
    ans += np.where(ans < 0, 2 * np.pi, 0.)
    return ans
np.arctan2 mapped to (0, 2 * pi)
dataset/ETHPy150Open mrterry/yoink/yoink/delta_e.py/_arctan2pi