admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-23 17:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: use string key for IS_INTERVENTION in complementary_info to avoid torch.load serialization error

Browse Source
This commit is contained in:
Khalil Meftah
2026-04-15 11:49:38 +02:00
parent 79a9ebdaa6
commit c88cf979f1
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/rl/actor.py
+1 -1
View File
@@ -338,7 +338,7 @@ def act_with_policy(
"discrete_penalty": torch.tensor( "discrete_penalty": torch.tensor(
[new_transition[TransitionKey.COMPLEMENTARY_DATA].get("discrete_penalty", 0.0)] [new_transition[TransitionKey.COMPLEMENTARY_DATA].get("discrete_penalty", 0.0)]
), ),
TeleopEvents.IS_INTERVENTION: is_intervention, "is_intervention": is_intervention,
} }
# Create transition for learner (convert to old format) # Create transition for learner (convert to old format)
list_transition_to_send_to_learner.append( list_transition_to_send_to_learner.append(