Refactor datasets into LeRobotDataset (#91)

Browse Source

Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>

...

This commit is contained in:

Remi

2024-04-25 12:23:12 +02:00

committed by GitHub

parent e760e4cd63

commit 659c69a1c0

90 changed files with 167 additions and 352 deletions

Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL

Download Patch File Download Diff File Expand all files Collapse all files

tests/data/aloha_sim_insertion_human/meta_data/episode_data_index.safetensors → tests/data/lerobot/aloha_sim_insertion_human/meta_data/episode_data_index.safetensors

tests/data/aloha_sim_insertion_human/meta_data/info.json → tests/data/lerobot/aloha_sim_insertion_human/meta_data/info.json

tests/data/aloha_sim_insertion_human/meta_data/stats.safetensors → tests/data/lerobot/aloha_sim_insertion_human/meta_data/stats.safetensors

tests/data/aloha_sim_insertion_human/stats.pth → tests/data/lerobot/aloha_sim_insertion_human/stats.pth

tests/data/aloha_sim_insertion_human/train/data-00000-of-00001.arrow → tests/data/lerobot/aloha_sim_insertion_human/train/data-00000-of-00001.arrow

tests/data/aloha_sim_insertion_human/train/dataset_info.json → tests/data/lerobot/aloha_sim_insertion_human/train/dataset_info.json

tests/data/aloha_sim_insertion_human/train/state.json → tests/data/lerobot/aloha_sim_insertion_human/train/state.json

tests/data/aloha_sim_insertion_scripted/meta_data/episode_data_index.safetensors → tests/data/lerobot/aloha_sim_insertion_scripted/meta_data/episode_data_index.safetensors

tests/data/aloha_sim_insertion_scripted/meta_data/info.json → tests/data/lerobot/aloha_sim_insertion_scripted/meta_data/info.json

tests/data/aloha_sim_insertion_scripted/meta_data/stats.safetensors → tests/data/lerobot/aloha_sim_insertion_scripted/meta_data/stats.safetensors

tests/data/aloha_sim_insertion_scripted/stats.pth → tests/data/lerobot/aloha_sim_insertion_scripted/stats.pth

tests/data/aloha_sim_insertion_scripted/train/data-00000-of-00001.arrow → tests/data/lerobot/aloha_sim_insertion_scripted/train/data-00000-of-00001.arrow

tests/data/aloha_sim_insertion_scripted/train/dataset_info.json → tests/data/lerobot/aloha_sim_insertion_scripted/train/dataset_info.json

tests/data/aloha_sim_insertion_scripted/train/state.json → tests/data/lerobot/aloha_sim_insertion_scripted/train/state.json

tests/data/aloha_sim_transfer_cube_human/meta_data/episode_data_index.safetensors → tests/data/lerobot/aloha_sim_transfer_cube_human/meta_data/episode_data_index.safetensors

tests/data/aloha_sim_transfer_cube_human/meta_data/info.json → tests/data/lerobot/aloha_sim_transfer_cube_human/meta_data/info.json

tests/data/aloha_sim_transfer_cube_human/meta_data/stats.safetensors → tests/data/lerobot/aloha_sim_transfer_cube_human/meta_data/stats.safetensors

tests/data/aloha_sim_transfer_cube_human/stats.pth → tests/data/lerobot/aloha_sim_transfer_cube_human/stats.pth

tests/data/aloha_sim_transfer_cube_human/train/data-00000-of-00001.arrow → tests/data/lerobot/aloha_sim_transfer_cube_human/train/data-00000-of-00001.arrow

tests/data/aloha_sim_transfer_cube_human/train/dataset_info.json → tests/data/lerobot/aloha_sim_transfer_cube_human/train/dataset_info.json

tests/data/aloha_sim_transfer_cube_human/train/state.json → tests/data/lerobot/aloha_sim_transfer_cube_human/train/state.json

tests/data/aloha_sim_transfer_cube_scripted/meta_data/episode_data_index.safetensors → tests/data/lerobot/aloha_sim_transfer_cube_scripted/meta_data/episode_data_index.safetensors

tests/data/aloha_sim_transfer_cube_scripted/meta_data/info.json → tests/data/lerobot/aloha_sim_transfer_cube_scripted/meta_data/info.json

tests/data/aloha_sim_transfer_cube_scripted/meta_data/stats.safetensors → tests/data/lerobot/aloha_sim_transfer_cube_scripted/meta_data/stats.safetensors

tests/data/aloha_sim_transfer_cube_scripted/stats.pth → tests/data/lerobot/aloha_sim_transfer_cube_scripted/stats.pth

tests/data/aloha_sim_transfer_cube_scripted/train/data-00000-of-00001.arrow → tests/data/lerobot/aloha_sim_transfer_cube_scripted/train/data-00000-of-00001.arrow

tests/data/aloha_sim_transfer_cube_scripted/train/dataset_info.json → tests/data/lerobot/aloha_sim_transfer_cube_scripted/train/dataset_info.json

tests/data/aloha_sim_transfer_cube_scripted/train/state.json → tests/data/lerobot/aloha_sim_transfer_cube_scripted/train/state.json

tests/data/pusht/meta_data/episode_data_index.safetensors → tests/data/lerobot/pusht/meta_data/episode_data_index.safetensors

tests/data/pusht/meta_data/info.json → tests/data/lerobot/pusht/meta_data/info.json

tests/data/pusht/meta_data/stats.safetensors → tests/data/lerobot/pusht/meta_data/stats.safetensors

tests/data/pusht/stats.pth → tests/data/lerobot/pusht/stats.pth

tests/data/pusht/train/data-00000-of-00001.arrow → tests/data/lerobot/pusht/train/data-00000-of-00001.arrow

tests/data/pusht/train/dataset_info.json → tests/data/lerobot/pusht/train/dataset_info.json

tests/data/pusht/train/meta_data/episode_data_index.safetensors → tests/data/lerobot/pusht/train/meta_data/episode_data_index.safetensors

tests/data/pusht/train/meta_data/info.json → tests/data/lerobot/pusht/train/meta_data/info.json

tests/data/pusht/train/meta_data/stats_action.safetensors → tests/data/lerobot/pusht/train/meta_data/stats_action.safetensors

tests/data/pusht/train/meta_data/stats_observation.image.safetensors → tests/data/lerobot/pusht/train/meta_data/stats_observation.image.safetensors

tests/data/pusht/train/meta_data/stats_observation.state.safetensors → tests/data/lerobot/pusht/train/meta_data/stats_observation.state.safetensors

tests/data/pusht/train/state.json → tests/data/lerobot/pusht/train/state.json

tests/data/xarm_lift_medium/meta_data/episode_data_index.safetensors → tests/data/lerobot/xarm_lift_medium/meta_data/episode_data_index.safetensors

tests/data/xarm_lift_medium/meta_data/info.json → tests/data/lerobot/xarm_lift_medium/meta_data/info.json

tests/data/xarm_lift_medium/meta_data/stats.safetensors → tests/data/lerobot/xarm_lift_medium/meta_data/stats.safetensors

tests/data/xarm_lift_medium/stats.pth → tests/data/lerobot/xarm_lift_medium/stats.pth

tests/data/xarm_lift_medium/train/data-00000-of-00001.arrow → tests/data/lerobot/xarm_lift_medium/train/data-00000-of-00001.arrow

tests/data/xarm_lift_medium/train/dataset_info.json → tests/data/lerobot/xarm_lift_medium/train/dataset_info.json

tests/data/xarm_lift_medium/train/state.json → tests/data/lerobot/xarm_lift_medium/train/state.json

tests/data/xarm_lift_medium_replay/meta_data/episode_data_index.safetensors → tests/data/lerobot/xarm_lift_medium_replay/meta_data/episode_data_index.safetensors

tests/data/xarm_lift_medium_replay/meta_data/info.json → tests/data/lerobot/xarm_lift_medium_replay/meta_data/info.json

tests/data/xarm_lift_medium_replay/meta_data/stats.safetensors → tests/data/lerobot/xarm_lift_medium_replay/meta_data/stats.safetensors

tests/data/xarm_lift_medium_replay/train/data-00000-of-00001.arrow → tests/data/lerobot/xarm_lift_medium_replay/train/data-00000-of-00001.arrow

tests/data/xarm_lift_medium_replay/train/dataset_info.json → tests/data/lerobot/xarm_lift_medium_replay/train/dataset_info.json

tests/data/xarm_lift_medium_replay/train/state.json → tests/data/lerobot/xarm_lift_medium_replay/train/state.json

tests/data/xarm_push_medium/meta_data/episode_data_index.safetensors → tests/data/lerobot/xarm_push_medium/meta_data/episode_data_index.safetensors

tests/data/xarm_push_medium/meta_data/info.json → tests/data/lerobot/xarm_push_medium/meta_data/info.json

tests/data/xarm_push_medium/meta_data/stats.safetensors → tests/data/lerobot/xarm_push_medium/meta_data/stats.safetensors

tests/data/xarm_push_medium/train/data-00000-of-00001.arrow → tests/data/lerobot/xarm_push_medium/train/data-00000-of-00001.arrow

tests/data/xarm_push_medium/train/dataset_info.json → tests/data/lerobot/xarm_push_medium/train/dataset_info.json

tests/data/xarm_push_medium/train/state.json → tests/data/lerobot/xarm_push_medium/train/state.json

tests/data/xarm_push_medium_replay/meta_data/episode_data_index.safetensors → tests/data/lerobot/xarm_push_medium_replay/meta_data/episode_data_index.safetensors

tests/data/xarm_push_medium_replay/meta_data/info.json → tests/data/lerobot/xarm_push_medium_replay/meta_data/info.json

tests/data/xarm_push_medium_replay/meta_data/stats.safetensors → tests/data/lerobot/xarm_push_medium_replay/meta_data/stats.safetensors

tests/data/xarm_push_medium_replay/train/data-00000-of-00001.arrow → tests/data/lerobot/xarm_push_medium_replay/train/data-00000-of-00001.arrow

tests/data/xarm_push_medium_replay/train/dataset_info.json → tests/data/lerobot/xarm_push_medium_replay/train/dataset_info.json

tests/data/xarm_push_medium_replay/train/state.json → tests/data/lerobot/xarm_push_medium_replay/train/state.json

tests/data/save_dataset_to_safetensors/pusht/frame_0.safetensors → tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_0.safetensors

tests/data/save_dataset_to_safetensors/pusht/frame_1.safetensors → tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_1.safetensors

tests/data/save_dataset_to_safetensors/pusht/frame_159.safetensors → tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_159.safetensors

tests/data/save_dataset_to_safetensors/pusht/frame_160.safetensors → tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_160.safetensors

tests/data/save_dataset_to_safetensors/pusht/frame_80.safetensors → tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_80.safetensors

tests/data/save_dataset_to_safetensors/pusht/frame_81.safetensors → tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_81.safetensors