From 82a2dc2415edde7114f6732c6afa6b5b36063855 Mon Sep 17 00:00:00 2001 From: Tavish Date: Sat, 8 Mar 2025 01:05:42 +0800 Subject: [PATCH] fix bugs when len of traj is 1 --- lerobot/common/datasets/v21/convert_stats.py | 1 + 1 file changed, 1 insertion(+) diff --git a/lerobot/common/datasets/v21/convert_stats.py b/lerobot/common/datasets/v21/convert_stats.py index 4a20b427..f0f49d77 100644 --- a/lerobot/common/datasets/v21/convert_stats.py +++ b/lerobot/common/datasets/v21/convert_stats.py @@ -40,6 +40,7 @@ def convert_episode_stats(dataset: LeRobotDataset, ep_idx: int): if ft["dtype"] == "video": # We sample only for videos ep_ft_data = sample_episode_video_frames(dataset, ep_idx, key) + ep_ft_data = ep_ft_data[None, ...] if ep_ft_data.ndim == 3 else ep_ft_data else: ep_ft_data = np.array(ep_data[key])