wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix bugs when len of traj is 1

This commit is contained in:
Tavish 2025-03-08 01:05:42 +08:00
parent 074f0ac8fe
commit 82a2dc2415
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
lerobot/common/datasets/v21/convert_stats.py
View File

@ -40,6 +40,7 @@ def convert_episode_stats(dataset: LeRobotDataset, ep_idx: int):
if ft["dtype"] == "video": if ft["dtype"] == "video":
# We sample only for videos # We sample only for videos
ep_ft_data = sample_episode_video_frames(dataset, ep_idx, key) 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: else:
ep_ft_data = np.array(ep_data[key]) ep_ft_data = np.array(ep_data[key])