From c3b8f65a8cfc4ea3cc69121c064850a46ef4abcf Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Tue, 18 Nov 2025 15:40:23 +0100 Subject: [PATCH] put axis-1 --- src/lerobot/processor/observation_processor.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/lerobot/processor/observation_processor.py b/src/lerobot/processor/observation_processor.py index 982bf9bf8..2142ae08c 100644 --- a/src/lerobot/processor/observation_processor.py +++ b/src/lerobot/processor/observation_processor.py @@ -251,7 +251,7 @@ class LiberoProcessorStep(ObservationProcessorStep): eef_axisangle = self._quat2axisangle(eef_quat) # (B, 3) # Concatenate into a single state vector - state = np.concatenate((eef_pos, eef_axisangle, gripper_qpos), axis=1) + state = np.concatenate((eef_pos, eef_axisangle, gripper_qpos), axis=-1) # Convert to tensor state_tensor = torch.from_numpy(state).float()