diff --git a/mini_bdx_runtime/mini_bdx_runtime/imu.py b/mini_bdx_runtime/mini_bdx_runtime/imu.py index 5fcb1fc..a645d30 100644 --- a/mini_bdx_runtime/mini_bdx_runtime/imu.py +++ b/mini_bdx_runtime/mini_bdx_runtime/imu.py @@ -119,6 +119,12 @@ class Imu: print("[IMU]:", e) continue + if gyro is None or raw_orientation is None: + continue + + if gyro.any() is None or raw_orientation.any() is None: + continue + # Converting to correct axes # euler = self.convert_axes(euler) euler[1] -= np.deg2rad(self.pitch_bias)