mirror of
https://github.com/huggingface/lerobot.git
synced 2026-05-24 13:09:43 +00:00
fix(rl): swap dict merge order to preserve teleop intervention flag (#3273)
Co-authored-by: Khalil Meftah <khalil.meftah@huggingface.co>
This commit is contained in:
@@ -551,8 +551,8 @@ def step_env_and_process_transition(
|
|||||||
terminated = terminated or processed_action_transition[TransitionKey.DONE]
|
terminated = terminated or processed_action_transition[TransitionKey.DONE]
|
||||||
truncated = truncated or processed_action_transition[TransitionKey.TRUNCATED]
|
truncated = truncated or processed_action_transition[TransitionKey.TRUNCATED]
|
||||||
complementary_data = processed_action_transition[TransitionKey.COMPLEMENTARY_DATA].copy()
|
complementary_data = processed_action_transition[TransitionKey.COMPLEMENTARY_DATA].copy()
|
||||||
new_info = processed_action_transition[TransitionKey.INFO].copy()
|
new_info = info.copy()
|
||||||
new_info.update(info)
|
new_info.update(processed_action_transition[TransitionKey.INFO])
|
||||||
|
|
||||||
new_transition = create_transition(
|
new_transition = create_transition(
|
||||||
observation=obs,
|
observation=obs,
|
||||||
|
|||||||
Reference in New Issue
Block a user