From a96845809be07edcc9914487502d92a3f3224e1b Mon Sep 17 00:00:00 2001 From: Tavish Date: Fri, 21 Feb 2025 18:50:49 +0800 Subject: [PATCH] fix berkeley_mvp_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 a66d1ce..c845fc6 100644 --- a/oxe_utils/transforms.py +++ b/oxe_utils/transforms.py @@ -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