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

Co-authored-by: Remi <re.cadene@gmail.com>
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
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d5883aa2c8ba2bcd8d047a77064112aa5d4c1c9b8595bb28935ec93ed53627e5
+oid sha256:52723265cba2ec839a5fcf75733813ecf91019ec0f7a49865fe233616e674583
 size 3056
--- a/tests/data/lerobot/pusht/train/data-00000-of-00001.arrow
+++ b/tests/data/lerobot/pusht/train/data-00000-of-00001.arrow
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0eab443dd492d0e271094290ae3cec2c9b2f4a19d35434eb5952cb37b0d40890
-size 18272
+oid sha256:8552d4ac6b618a5b2741e174d51f1d4fc0e5f4e6cc7026bebdb6ed145373b042
+size 18320
--- a/tests/data/lerobot/pusht/train/state.json
+++ b/tests/data/lerobot/pusht/train/state.json
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8c1a72239bb56a6c5714f18d849557c89feb858840e8f86689d017bb49551379
+oid sha256:a522c7815565f1f81a8bb5a853263405ab8c3b087ecbc7a3b004848891d77342
 size 247
--- a/tests/data/lerobot/pusht_keypoints/meta_data/episode_data_index.safetensors
+++ b/tests/data/lerobot/pusht_keypoints/meta_data/episode_data_index.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a1cd3db853d0f92e1696fe297c550200219d85befdeb5b5eacae4b10a74d9896
+size 136
--- a/tests/data/lerobot/pusht_keypoints/meta_data/info.json
+++ b/tests/data/lerobot/pusht_keypoints/meta_data/info.json
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:dbf25de102227dd2d8c3b6c61e1fc25a026d44f151161b88bc9a9eb101e942e4
+size 33
--- a/tests/data/lerobot/pusht_keypoints/meta_data/stats.safetensors
+++ b/tests/data/lerobot/pusht_keypoints/meta_data/stats.safetensors
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:50b3c026da835560f9b87e7dfd28673e766bfb58d56c85002687d0a599b6fa43
+size 3304
--- a/tests/data/lerobot/pusht_keypoints/train/data-00000-of-00001.arrow
+++ b/tests/data/lerobot/pusht_keypoints/train/data-00000-of-00001.arrow
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:958798d23a1690449744961f8c3ed934efe950c664e5fd729468959362840218
+size 20336
--- a/tests/data/lerobot/pusht_keypoints/train/dataset_info.json
+++ b/tests/data/lerobot/pusht_keypoints/train/dataset_info.json
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:686d9d9bad8815d67597b997058d9853a04e5bdbe4eed038f4da9806f867af3d
+size 1098
--- a/tests/data/lerobot/pusht_keypoints/train/state.json
+++ b/tests/data/lerobot/pusht_keypoints/train/state.json
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:f22ee3500aca1bea0afdda429e841c57a3278dfea92c79bbbf5dac5f984ed648
+size 247
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_0.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_0.safetensors
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d9cc073bcb335024500fe7c823f142a3b4f038ff458d8c47fb6a6918f8f6d5fd
+oid sha256:b99bbb7332557d47b108fd0262d911c99f5bfce30fa3e76dc802b927284135e7
 size 111338
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_1.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_1.safetensors
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:58c50ef6413b6b3acb7ad280281cdd4eba553f7d3d0b4dad20c262025d610f2b
+oid sha256:0f63430455e1ca7a5fe28c81a15fc0eb82758035e6b3d623e7e7952e71cb262a
 size 111338
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_159.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_159.safetensors
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bd1d26e983e2910ec170cd6ac1f4de4d7cb447ee24b516a74f42765d4894e048
+oid sha256:0b88c39db5b13da646fd5876bd765213569387591d30ec665d048ae1070db0b9
 size 111338
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_160.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_160.safetensors
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e1247a9d4683520ed338f3fd410cc200999e4b82da573cd499095ba02037586f
+oid sha256:68eb245890f9537851ea7fb227472dcd4f1fa3820a7c3294a4989e2b9896d078
 size 111338
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_80.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_80.safetensors
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b24f3c3d41428b768082eb3b02b5e22dc9540aa4dbe756d43be214d51e97adba
+oid sha256:00c74e17bbf7d428b0b0869f388d348820a938c417b3c888a1384980bb53d4d0
 size 111338
--- a/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_81.safetensors
+++ b/tests/data/save_dataset_to_safetensors/lerobot/pusht/frame_81.safetensors
@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:5301dc61b585fbfbdb6ce681ffcd52fc26b64a3767567c228a9e4404f7bcb926
+oid sha256:a5a7f66704640ba18f756fc44c00721c77a406f412a3a9fcc1a2b1868c978444
 size 111338
--- 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
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1ae7f6a7f4ee8340ec73b0e7f1e167046af2af0a22381e0cd3ff42f311e098e0
+size 794
--- 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
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:2eeb1b185b505450f8a2b6042537d65d2d8f5ee1396cf878a50d3d2aa3a22822
+size 794
--- 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
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7f2bb24887f9d4c49ad562429f419b7b66f4310a59877104a98d3c5c6ddca996
+size 794
--- 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
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:a52fe583c816fdfb962111dd1ee1c113a5f4b9699246fab8648f89e056979f8e
+size 794
--- 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
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:70dbf161581b860e255573eb1ef90f4defd134d8dcf0afea16099c859c4a8f85
+size 794
--- 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
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:198abd0ec4231c13cadf707d553cba3860acbc74a073406ed184eab5495acdfa
+size 794