Include observation.environment_state with keypoints in PushT dataset (#303)

Co-authored-by: Remi <re.cadene@gmail.com>
2026-06-05 05:11:25 +00:00 · 2024-07-09 08:27:40 +01:00
parent 7bd5ab16d1
commit a4d77b99f0
27 changed files with 739 additions and 609 deletions
--- a/tests/data/lerobot/pusht/meta_data/stats.safetensors
+++ b/tests/data/lerobot/pusht/meta_data/stats.safetensors
--- a/tests/data/lerobot/pusht/train/data-00000-of-00001.arrow
+++ b/tests/data/lerobot/pusht/train/data-00000-of-00001.arrow
--- a/tests/data/lerobot/pusht/train/state.json
+++ b/tests/data/lerobot/pusht/train/state.json
--- a/tests/data/lerobot/pusht_keypoints/meta_data/episode_data_index.safetensors
+++ b/tests/data/lerobot/pusht_keypoints/meta_data/episode_data_index.safetensors
--- a/tests/data/lerobot/pusht_keypoints/meta_data/info.json
+++ b/tests/data/lerobot/pusht_keypoints/meta_data/info.json
--- a/tests/data/lerobot/pusht_keypoints/meta_data/stats.safetensors
+++ b/tests/data/lerobot/pusht_keypoints/meta_data/stats.safetensors
--- a/tests/data/lerobot/pusht_keypoints/train/data-00000-of-00001.arrow
+++ b/tests/data/lerobot/pusht_keypoints/train/data-00000-of-00001.arrow
--- a/tests/data/lerobot/pusht_keypoints/train/dataset_info.json
+++ b/tests/data/lerobot/pusht_keypoints/train/dataset_info.json
--- a/tests/data/lerobot/pusht_keypoints/train/state.json
+++ b/tests/data/lerobot/pusht_keypoints/train/state.json
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_0.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_0.safetensors
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_1.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_1.safetensors
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_159.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_159.safetensors
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_160.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_160.safetensors
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_80.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_80.safetensors
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_81.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_81.safetensors
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht_keypoints/frame_0.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht_keypoints/frame_0.safetensors
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht_keypoints/frame_1.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht_keypoints/frame_1.safetensors
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht_keypoints/frame_159.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht_keypoints/frame_159.safetensors
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht_keypoints/frame_160.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht_keypoints/frame_160.safetensors
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht_keypoints/frame_80.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht_keypoints/frame_80.safetensors
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht_keypoints/frame_81.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht_keypoints/frame_81.safetensors