From 6ab4ac3202ab43642af723b2dcba3a938dd48338 Mon Sep 17 00:00:00 2001 From: Tavish Date: Thu, 20 Feb 2025 20:54:56 +0800 Subject: [PATCH] fix berkeley_rpt_dataset_transform --- oxe_utils/transforms.py | 1 + 1 file changed, 1 insertion(+) diff --git a/oxe_utils/transforms.py b/oxe_utils/transforms.py index 2d97c04..a66d1ce 100644 --- a/oxe_utils/transforms.py +++ b/oxe_utils/transforms.py @@ -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]: + trajectory["observation"]["gripper"] = trajectory["observation"]["gripper"][:, None] return trajectory