diffusion_policy/dataset/mujoco_image_dataset.py may have error

[tainnanlshuf-cpu]

diffusion_policy/dataset/mujoco_image_dataset.py line 64
'agent_pos': np.concatenate([self.replay_buffer['robot_0_tcp_xyz_wxyz'], self.replay_buffer['robot_0_tcp_xyz_wxyz']], axis=-1) may be wrong
and change to this
'agent_pos': np.concatenate([self.replay_buffer['robot_0_tcp_xyz_wxyz'], self.replay_buffer['robot_0_gripper_width']], axis=-1)