diff --git a/habitat-lab/habitat/tasks/nav/nav.py b/habitat-lab/habitat/tasks/nav/nav.py index edf759f0a3..5b9b18deba 100644 --- a/habitat-lab/habitat/tasks/nav/nav.py +++ b/habitat-lab/habitat/tasks/nav/nav.py @@ -356,6 +356,7 @@ def _quat_to_xy_heading(self, quat): heading_vector = quaternion_rotate_vector(quat, direction_vector) phi = cartesian_to_polar(-heading_vector[2], heading_vector[0])[1] + phi = np.clip(phi, -np.pi, np.pi) return np.array([phi], dtype=np.float32) def get_observation(