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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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

0

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