ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-31 19:01:28 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: include IS_INTERVENTION in complementary_info sent to learner for offline replay buffer

Browse Source
This commit is contained in:
Khalil Meftah
2026-04-14 16:35:08 +02:00
parent e6d282108d
commit 64dc08cb7b
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/lerobot/rl/actor.py
View File

@@ -327,7 +327,8 @@ def act_with_policy(
# Check for intervention from transition info
intervention_info = new_transition[TransitionKey.INFO]
if intervention_info.get(TeleopEvents.IS_INTERVENTION, False):
is_intervention = intervention_info.get(TeleopEvents.IS_INTERVENTION, False)
if is_intervention:
episode_intervention = True
episode_intervention_steps += 1
@@ -335,6 +336,7 @@ def act_with_policy(
"discrete_penalty": torch.tensor(
[new_transition[TransitionKey.COMPLEMENTARY_DATA].get("discrete_penalty", 0.0)]
),
TeleopEvents.IS_INTERVENTION: is_intervention,
}
# Create transition for learner (convert to old format)
list_transition_to_send_to_learner.append(