From 5cfca59ec79783b2fb9a125d8d74f911698a49b9 Mon Sep 17 00:00:00 2001
From: Khalil Meftah <khalil.meftah@huggingface.co>
Date: Thu, 21 May 2026 11:05:02 +0200
Subject: [PATCH] fix(rewards/topreward): add missing input keys
 mm_token_type_ids

---
 src/lerobot/rewards/topreward/processor_topreward.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/lerobot/rewards/topreward/processor_topreward.py b/src/lerobot/rewards/topreward/processor_topreward.py
index bd85e541b..160f87fc5 100644
--- a/src/lerobot/rewards/topreward/processor_topreward.py
+++ b/src/lerobot/rewards/topreward/processor_topreward.py
@@ -65,6 +65,7 @@ TOPREWARD_VLM_INPUT_KEYS = (
     "image_grid_thw",
     "video_grid_thw",
     "second_per_grid_ts",
+    "mm_token_type_ids",
 )
 TOPREWARD_METADATA_KEYS = ("prompt_length",)
 TOPREWARD_INPUT_KEYS = TOPREWARD_VLM_INPUT_KEYS + TOPREWARD_METADATA_KEYS