admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-23 09:46:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

put axis-1

Browse Source
This commit is contained in:
Jade Choghari
2025-11-18 15:40:23 +01:00
parent 33a8d31af0
commit c3b8f65a8c
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/processor/observation_processor.py
+1 -1
View File
@@ -251,7 +251,7 @@ class LiberoProcessorStep(ObservationProcessorStep):
eef_axisangle = self._quat2axisangle(eef_quat) # (B, 3) eef_axisangle = self._quat2axisangle(eef_quat) # (B, 3)
# Concatenate into a single state vector # 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 # Convert to tensor
state_tensor = torch.from_numpy(state).float() state_tensor = torch.from_numpy(state).float()