ydy0615/lerobot-clone
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-06-03 20:31:25 +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

2
src/lerobot/processor/observation_processor.py
View File

@@ -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()