admin/any4lerobot
1
0
Fork 0
mirror of https://github.com/Tavish9/any4lerobot.git synced 2026-05-30 12:49:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix berkeley_rpt_dataset_transform

Browse Source
This commit is contained in:
Tavish
2025-02-20 20:54:56 +08:00
parent 6e7bdc3e9e
commit 6ab4ac3202
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
@@ -621,6 +621,7 @@ def berkeley_mvp_dataset_transform(trajectory: Dict[str, Any]) -> Dict[str, Any]
def berkeley_rpt_dataset_transform(trajectory: Dict[str, Any]) -> Dict[str, Any]: def berkeley_rpt_dataset_transform(trajectory: Dict[str, Any]) -> Dict[str, Any]:
trajectory["observation"]["gripper"] = trajectory["observation"]["gripper"][:, None]
return trajectory return trajectory