diff --git a/src/lerobot/rl/gym_manipulator.py b/src/lerobot/rl/gym_manipulator.py index bd64d205f8..dd3ae44c43 100644 --- a/src/lerobot/rl/gym_manipulator.py +++ b/src/lerobot/rl/gym_manipulator.py @@ -551,8 +551,8 @@ def step_env_and_process_transition( terminated = terminated or processed_action_transition[TransitionKey.DONE] truncated = truncated or processed_action_transition[TransitionKey.TRUNCATED] complementary_data = processed_action_transition[TransitionKey.COMPLEMENTARY_DATA].copy() - new_info = processed_action_transition[TransitionKey.INFO].copy() - new_info.update(info) + new_info = info.copy() + new_info.update(processed_action_transition[TransitionKey.INFO]) new_transition = create_transition( observation=obs,