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

This commit is contained in:
Khalil Meftah
2026-04-15 11:49:38 +02:00
parent 79a9ebdaa6
commit c88cf979f1

View File

@@ -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(