def cart2pol()

in robosumo/envs/utils.py [0:0]

• 4 lines of code
• 1 McCabe index (conditional complexity)

def cart2pol(vec):
    """Convert a cartesian 2D vector to polar coordinates."""
    r = np.sqrt(vec[0]**2 + vec[1]**2)
    theta = np.arctan2(vec[1], vec[0])
    return np.asarray([r, theta])