Commit Graph

805 Commits

Author SHA1 Message Date
Simon Alibert fb73cdb9a4 Update dataset doc 2024-10-23 00:32:28 +02:00
Simon Alibert 9dca233d7e Fix episode chunk 2024-10-23 00:27:14 +02:00
Simon Alibert c3c0141738 Update & fix conversion script 2024-10-23 00:05:31 +02:00
Simon Alibert c72dc23c43 Remove total_episodes from default parquet path 2024-10-23 00:03:30 +02:00
Simon Alibert 237a484be0 Fix paths & add add_frame doc 2024-10-22 22:46:34 +02:00
Simon Alibert 6c2cb6e107 Remove populate dataset 2024-10-22 20:21:26 +02:00
Simon Alibert b46db7ea73 Fix tests 2024-10-22 20:14:06 +02:00
Simon Alibert ee52b8b782 Add channels to intelrealsense 2024-10-22 20:07:11 +02:00
Simon Alibert a805458c7e Add local_files_only, encode_videos, fix bugs to pass tests (WIP) 2024-10-22 19:57:52 +02:00
Simon Alibert e991a31061 Improve consistency between __init__() and create(), WIP on consolidate 2024-10-22 00:19:25 +02:00
Simon Alibert c4c0a43de7 add delete_episode, WIP on consolidate 2024-10-21 20:10:13 +02:00
Simon Alibert 299451af81 Add add_episode & task logic 2024-10-21 19:30:20 +02:00
Simon Alibert 9ebf8b88ec Merge remote-tracking branch 'origin/main' into user/aliberts/2024_09_25_reshape_dataset 2024-10-21 00:22:30 +02:00
Simon Alibert c1232a01e2 Add add_frame, empty dataset creation 2024-10-21 00:16:52 +02:00
Simon Alibert 3b925c3dce Add ImageWriter 2024-10-21 00:15:09 +02:00
Simon Alibert e46bdb9d30 Change card creation 2024-10-20 14:01:10 +02:00
Simon Alibert 9316cf46ef Add file paths 2024-10-20 14:00:19 +02:00
Simon Alibert ac3798bd62 Move default paths, use jsonlines for tasks 2024-10-18 17:53:25 +02:00
Simon Alibert bce3dc3bfa Add load_metadata 2024-10-18 14:59:09 +02:00
Simon Alibert 1a51505ec6 Add download_metadata, move default paths 2024-10-18 14:48:34 +02:00
Simon Alibert e7355ba595 Fix episodes.jsonl 2024-10-18 12:03:29 +02:00
Boris Zimka c351e1fff9
Fix gymnasium version as pre-1.0.0 (#471)
Co-authored-by: Remi <re.cadene@gmail.com>
Co-authored-by: Remi <remi.cadene@huggingface.co>
2024-10-18 10:23:27 +02:00
Simon Alibert 91e8ce772b Remove caret requirement 2024-10-18 09:48:38 +02:00
Simon Alibert 3a9f964429 Add copyrights 2024-10-18 00:31:21 +02:00
Simon Alibert beacb7e957 Cleanup 2024-10-18 00:30:16 +02:00
Simon Alibert be64d54bd9 Update doc 2024-10-18 00:29:50 +02:00
Simon Alibert 354f37aed3 Merge remote-tracking branch 'origin/main' into user/aliberts/2024_09_25_reshape_dataset 2024-10-17 23:34:00 +02:00
Simon Alibert d0d8193625 Add unitreeh and umi 2024-10-17 23:33:51 +02:00
Alexander Soare cd0fc261c0
Make `say(blocking=True)` work for Linux (#460) 2024-10-17 15:22:21 +01:00
Simon Alibert 7242c57400 Cleanup 2024-10-17 16:08:37 +02:00
Simon Alibert 3ee3739edc Add batch conversion log 2024-10-17 13:08:58 +02:00
Simon Alibert ad3f112d16 Add fixes for lfs tracking 2024-10-17 12:58:48 +02:00
Simon Alibert 50a75ad3fe Write episodes as jsonlines 2024-10-17 10:17:27 +02:00
Simon Alibert c146ba936f Add episode chunks logic, move_videos & lfs tracking fix 2024-10-16 23:34:54 +02:00
Remi 77478d50e5
Refactor `record` with `add_frame` (#468)
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
2024-10-16 20:51:35 +02:00
Simon Alibert 110264000f Add fixes for batch convert 2024-10-15 19:03:11 +02:00
Simon Alibert 9433ac52ec WIP add batch convert 2024-10-15 13:08:31 +02:00
Simon Alibert da78bbfd16 Update load_tasks doc 2024-10-15 11:06:28 +02:00
Simon Alibert 835ab5a81b Cleanup, fix load_tasks 2024-10-15 11:05:16 +02:00
Simon Alibert f96773de10 Fix safe_version 2024-10-14 13:51:40 +02:00
Simon Alibert cbc51e1341 Extend v1 compatibility 2024-10-14 10:14:27 +02:00
Simon Alibert cf633344be Add multitask support, refactor conversion script 2024-10-13 21:21:40 +02:00
Simon Alibert 8bd406e607 Add suggestions from code review 2024-10-11 18:52:11 +02:00
Simon Alibert 3ea53124e0 Add padding keys and download_data option 2024-10-11 17:38:47 +02:00
Simon Alibert 7f680886b0 Add huggingface-hub patch for offline snapshot_download with local_dir 2024-10-11 11:03:11 +02:00
Simon Alibert 6d2bc11365 Add doc, scrap video_frame_keys attribute 2024-10-11 10:59:38 +02:00
Simon Alibert b417cebc4e Update LeRobotDataset.__get_item__ 2024-10-10 21:32:14 +02:00
Remi 97b1feb0b3
Add policy/act_aloha_real.yaml + env/act_real.yaml (#429)
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
2024-10-10 17:12:45 +02:00
Simon Alibert 3113038beb Merge remote-tracking branch 'origin/main' into user/aliberts/2024_09_25_reshape_dataset 2024-10-09 14:33:58 +02:00
Simon Alibert 096824b5ff Rework LeRobotDataset.__init__ 2024-10-09 14:33:26 +02:00