From c88cf979f11a58ff22381cbfd8a9a844ffc6a565 Mon Sep 17 00:00:00 2001 From: Khalil Meftah Date: Wed, 15 Apr 2026 11:49:38 +0200 Subject: [PATCH] fix: use string key for IS_INTERVENTION in complementary_info to avoid torch.load serialization error --- src/lerobot/rl/actor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/rl/actor.py b/src/lerobot/rl/actor.py index 6d4b3022e..83c88fa96 100644 --- a/src/lerobot/rl/actor.py +++ b/src/lerobot/rl/actor.py @@ -338,7 +338,7 @@ def act_with_policy( "discrete_penalty": torch.tensor( [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) list_transition_to_send_to_learner.append(