admin/any4lerobot
1
0
Fork 0
mirror of https://github.com/Tavish9/any4lerobot.git synced 2026-05-18 07:29:44 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix berkeley_mvp_dataset_transform

Browse Source
This commit is contained in:
Tavish
2025-02-21 18:50:49 +08:00
parent 36cf315bef
commit a96845809b
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
oxe_utils/transforms.py
+1
View File
@@ -617,6 +617,7 @@ def robo_net_dataset_transform(trajectory: Dict[str, Any]) -> Dict[str, Any]:
def berkeley_mvp_dataset_transform(trajectory: Dict[str, Any]) -> Dict[str, Any]:
trajectory["observation"]["gripper"] = trajectory["observation"]["gripper"][:, None]
return trajectory