Add rne_postconstraint dependence for FRAMELINACC and FRAMEANGACC sensors to MJX.

PiperOrigin-RevId: 746009774
Change-Id: Ide45298b463f1ad04f1ffa7f60a7a8aeaf18f063

mjx/mujoco/mjx/_src/io.py

@@ -129,6 +129,8 @@ def put_model(
       np.any(m.sensor_type == types.SensorType.ACCELEROMETER)
       | np.any(m.sensor_type == types.SensorType.FORCE)
       | np.any(m.sensor_type == types.SensorType.TORQUE)
+      | np.any(m.sensor_type == types.SensorType.FRAMELINACC)
+      | np.any(m.sensor_type == types.SensorType.FRAMEANGACC)
   )
   eq_connect_weld = np.any(m.eq_type == types.EqType.CONNECT) | np.any(
       m.eq_type == types.EqType.WELD

mjx/mujoco/mjx/_src/sensor.py

@@ -441,6 +441,8 @@ def sensor_acc(m: Model, d: Data) -> Data:
       SensorType.ACCELEROMETER,
       SensorType.FORCE,
       SensorType.TORQUE,
+      SensorType.FRAMELINACC,
+      SensorType.FRAMEANGACC,
   }:
     d = smooth.rne_postconstraint(m, d)