Dimension of "ee_state"

[Valarzz]

Hi, I want to ask how I can get the corresponding "ee_states" in the dataset from the online rollout.

In the dataset, "data/demo_{episode_idx}/obs/ee_states" has the shape of 6; In online rollout, obs = env.set_init_state(init_states_), obs["robot0_eef_pos"] has the shape of 3, and obs["robot0_eef_quat"] has the shape of 4.

Does "ee_states" includes both "robot0_eef_pos" and "robot0_eef_quat"? If yes, why there's one more dimension? Which dimension should I abandon? If no, how can I get "ee_states" from obs?

Thanks!