From 62f716dbc904f67724301d7a30c9b160c107dac0 Mon Sep 17 00:00:00 2001
From: Michel Aractingi <michel.aractingi@huggingface.co>
Date: Mon, 11 Aug 2025 19:02:33 +0200
Subject: [PATCH] nit in delta action

---
 src/lerobot/processor/delta_action_processor.py | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/src/lerobot/processor/delta_action_processor.py b/src/lerobot/processor/delta_action_processor.py
index 63a8b717f..1b644acde 100644
--- a/src/lerobot/processor/delta_action_processor.py
+++ b/src/lerobot/processor/delta_action_processor.py
@@ -36,12 +36,14 @@ class MapTensorToDeltaActionDict(ActionProcessor):
             action = action.squeeze(0)
 
         # TODO (maractingi): add rotation
-        return {
+        delta_action = {
             "action.delta_x": action[0],
             "action.delta_y": action[1],
             "action.delta_z": action[2],
-            "action.gripper": action[3],
         }
+        if action.shape[0] > 3:
+            delta_action["action.gripper"] = action[3]
+        return delta_action
 
 
 @ProcessorStepRegistry.register("map_delta_action_to_robot_action")