Commit Graph

761 Commits

Author SHA1 Message Date
Simon Alibert aed9f4036a Refactor dataset features 2024-11-05 13:10:43 +01:00
Simon Alibert 757ea175d3 Fix test_examples 2024-11-03 19:41:50 +01:00
Simon Alibert b69a132737 Fix test_examples 2024-11-03 19:30:56 +01:00
Simon Alibert 7b159a6b22 Move calculate_episode_data_index 2024-11-03 19:13:00 +01:00
Simon Alibert 74270c8c91 Remove reset_episode_index 2024-11-03 19:07:43 +01:00
Simon Alibert a6762ec316 Fix advanced example 2 2024-11-03 19:03:15 +01:00
Simon Alibert fde29e0167 Merge remote-tracking branch 'origin/main' into user/aliberts/2024_09_25_reshape_dataset 2024-11-03 18:32:14 +01:00
Simon Alibert 56e4603d5b Deactivate policies backward compatibility test 2024-11-03 18:30:50 +01:00
Simon Alibert f6c90ca35f Remove end-to-end tests 2024-11-03 18:28:54 +01:00
Simon Alibert c2d6fb6119 Fix werkzeug alert 2024-11-03 18:23:38 +01:00
Simon Alibert 95a4b59b5b Fix vizualize 2024-11-03 18:18:23 +01:00
Simon Alibert 16103cb8b7 Fix hanging 2024-11-03 18:07:55 +01:00
Simon Alibert e4ba084e25 Add LeRobotDatasetMetadata 2024-11-03 18:07:37 +01:00
Simon Alibert ac79e8cb36 Fix test_online_buffer.py 2024-11-03 13:15:01 +01:00
Simon Alibert df2cb51364 Add todo in skipped test 2024-11-02 20:01:34 +01:00
Simon Alibert 7a342db9c4 Add more options to img factories 2024-11-02 20:01:02 +01:00
Simon Alibert 6b2ec1ed77 Add test_image_writer, accept PIL images, improve ImageWriter perf in main process 2024-11-02 20:00:07 +01:00
Simon Alibert 375abd3020 Add img and img_tensor factories 2024-11-02 13:06:38 +01:00
Simon Alibert 293bdc7f67 Simplify, add test content, add todo 2024-11-01 19:55:28 +01:00
Simon Alibert 79d114cc1f Rename num_samples -> num_frames for consistency 2024-11-01 19:47:16 +01:00
Simon Alibert 2650872b76 Add tasks and episodes factories 2024-11-01 13:37:17 +01:00
Simon Alibert cd1509d805 Mock snapshot_download 2024-11-01 10:58:09 +01:00
Simon Alibert 5ea7c78237 Remove obsolete code 2024-10-31 21:43:57 +01:00
Simon Alibert 443a9eec88 Remove/comment obsolete tests 2024-10-31 21:43:29 +01:00
Simon Alibert ab23a4fd27 Add fixtures in test_datasets 2024-10-31 21:32:09 +01:00
Simon Alibert 1267c3e955 Split fixtures into factories and files 2024-10-31 21:29:50 +01:00
Arsen Ohanyan e0df56de62
Fix config file (#495) 2024-10-31 16:41:49 +01:00
Simon Alibert c70b8d0abc Update doc 2024-10-31 14:08:13 +01:00
Simon Alibert e1845d4dcc Update doc 2024-10-31 14:06:54 +01:00
Simon Alibert e69f0c5059 Add test_delta_timestamps.py 2024-10-31 13:48:40 +01:00
Simon Alibert ff84024ee9 Add dataset fixtures 2024-10-31 13:46:46 +01:00
Hirokazu Ishida 538455a965
feat: enable to use multiple rgb encoders per camera in diffusion policy (#484)
Co-authored-by: Alexander Soare <alexander.soare159@gmail.com>
2024-10-30 11:00:05 +01:00
Simon Alibert ee51f54cb5 Remove dataset from image_transform tests 2024-10-29 16:08:01 +01:00
Simon Alibert fee5fa5c2e Remove image_writer arg 2024-10-29 16:07:10 +01:00
Simon Alibert 8d57093ce7 Merge remote-tracking branch 'origin/main' into user/aliberts/2024_09_25_reshape_dataset 2024-10-29 15:25:25 +01:00
Simon Alibert 4c22de20a6 Add sanity check 2024-10-28 12:26:07 +01:00
Simon Alibert 51e87f6f97 Fix image writer 2024-10-28 12:01:32 +01:00
Remi 172809a502
[Fix] Move back to manual calibration (#488) 2024-10-26 15:27:21 +02:00
Remi 55e4ff6742
Fix autocalib moss (#486) 2024-10-26 12:15:17 +02:00
Simon Alibert df3d2ec5df Speedup test 2024-10-26 10:59:06 +02:00
Simon Alibert e210d795de Add video_info, fix image_writer 2024-10-25 16:55:33 +02:00
Remi 07e8716315
Add FeetechMotorsBus, SO-100, Moss-v1 (#419)
Co-authored-by: jess-moss <jess.moss@huggingface.co>
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
2024-10-25 11:23:55 +02:00
Simon Alibert 18ffa4248b Add json/jsonl io functions 2024-10-24 11:49:53 +02:00
Simon Alibert 8bcf81fa24 Add todo 2024-10-24 11:38:32 +02:00
Simon Alibert 615894d3fb Add test_same_attributes_defined 2024-10-24 11:37:44 +02:00
Simon Alibert 450eae310b Add error msg 2024-10-24 00:13:53 +02:00
Simon Alibert 60865e8980 Allow dataset creation without robot 2024-10-24 00:13:21 +02:00
Simon Alibert 0d77be90ee Move ImageWriter creation inside the dataset 2024-10-23 23:12:44 +02:00
Simon Alibert 0098bd264e Nits 2024-10-23 20:55:54 +02:00
Simon Alibert 1aba80d93f Fix consolidate 2024-10-23 18:45:59 +02:00