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

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

diff --git a/src/lerobot/rewards/robometer/modeling_robometer.py b/src/lerobot/rewards/robometer/modeling_robometer.py
index 6f462de85..aea49deae 100644
--- a/src/lerobot/rewards/robometer/modeling_robometer.py
+++ b/src/lerobot/rewards/robometer/modeling_robometer.py
@@ -78,6 +78,7 @@ ROBOMETER_QWEN_INPUT_KEYS = (
     "image_grid_thw",
     "video_grid_thw",
     "second_per_grid_ts",
+    "mm_token_type_ids",
 )
 ROBOMETER_METADATA_KEYS = (
     "prog_token_id",