From 1ede000bdd9c19e130bdac1f81580e4ccb713266 Mon Sep 17 00:00:00 2001 From: Radu <218564757+remete618@users.noreply.github.com> Date: Tue, 14 Apr 2026 07:20:54 -0700 Subject: [PATCH] fix(rl): swap dict merge order to preserve teleop intervention flag (#3273) Co-authored-by: Khalil Meftah --- src/lerobot/rl/gym_manipulator.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/lerobot/rl/gym_manipulator.py b/src/lerobot/rl/gym_manipulator.py index b6ff7155a..2190070f5 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,