diff --git a/tests/data/aloha_sim_insertion_human/meta_data/episode_data_index.safetensors b/tests/data/aloha_sim_insertion_human/meta_data/episode_data_index.safetensors new file mode 100644 index 000000000..8ed156c4b Binary files /dev/null and b/tests/data/aloha_sim_insertion_human/meta_data/episode_data_index.safetensors differ diff --git a/tests/data/aloha_sim_insertion_human/meta_data/info.json b/tests/data/aloha_sim_insertion_human/meta_data/info.json new file mode 100644 index 000000000..02e62b67d --- /dev/null +++ b/tests/data/aloha_sim_insertion_human/meta_data/info.json @@ -0,0 +1,3 @@ +{ + "fps": 50 +} \ No newline at end of file diff --git a/tests/data/aloha_sim_insertion_human/meta_data/stats.safetensors b/tests/data/aloha_sim_insertion_human/meta_data/stats.safetensors new file mode 100644 index 000000000..e95645033 Binary files /dev/null and b/tests/data/aloha_sim_insertion_human/meta_data/stats.safetensors differ diff --git a/tests/data/aloha_sim_insertion_human/train/data-00000-of-00001.arrow b/tests/data/aloha_sim_insertion_human/train/data-00000-of-00001.arrow index 165298cfb..f6c89ff6d 100644 Binary files a/tests/data/aloha_sim_insertion_human/train/data-00000-of-00001.arrow and b/tests/data/aloha_sim_insertion_human/train/data-00000-of-00001.arrow differ diff --git a/tests/data/aloha_sim_insertion_human/train/dataset_info.json b/tests/data/aloha_sim_insertion_human/train/dataset_info.json index 542c7bf12..69f80830c 100644 --- a/tests/data/aloha_sim_insertion_human/train/dataset_info.json +++ b/tests/data/aloha_sim_insertion_human/train/dataset_info.json @@ -21,11 +21,11 @@ "length": 14, "_type": "Sequence" }, - "episode_id": { + "episode_index": { "dtype": "int64", "_type": "Value" }, - "frame_id": { + "frame_index": { "dtype": "int64", "_type": "Value" }, @@ -37,14 +37,6 @@ "dtype": "bool", "_type": "Value" }, - "episode_data_index_from": { - "dtype": "int64", - "_type": "Value" - }, - "episode_data_index_to": { - "dtype": "int64", - "_type": "Value" - }, "index": { "dtype": "int64", "_type": "Value" diff --git a/tests/data/aloha_sim_insertion_human/train/state.json b/tests/data/aloha_sim_insertion_human/train/state.json index 39101fd57..153a4123b 100644 --- a/tests/data/aloha_sim_insertion_human/train/state.json +++ b/tests/data/aloha_sim_insertion_human/train/state.json @@ -4,7 +4,7 @@ "filename": "data-00000-of-00001.arrow" } ], - "_fingerprint": "d79cf82ffc86f110", + "_fingerprint": "22eeca7a3f4725ee", "_format_columns": null, "_format_kwargs": {}, "_format_type": "torch", diff --git a/tests/data/aloha_sim_insertion_scripted/meta_data/episode_data_index.safetensors b/tests/data/aloha_sim_insertion_scripted/meta_data/episode_data_index.safetensors new file mode 100644 index 000000000..8685de718 Binary files /dev/null and b/tests/data/aloha_sim_insertion_scripted/meta_data/episode_data_index.safetensors differ diff --git a/tests/data/aloha_sim_insertion_scripted/meta_data/info.json b/tests/data/aloha_sim_insertion_scripted/meta_data/info.json new file mode 100644 index 000000000..02e62b67d --- /dev/null +++ b/tests/data/aloha_sim_insertion_scripted/meta_data/info.json @@ -0,0 +1,3 @@ +{ + "fps": 50 +} \ No newline at end of file diff --git a/tests/data/aloha_sim_insertion_scripted/meta_data/stats.safetensors b/tests/data/aloha_sim_insertion_scripted/meta_data/stats.safetensors new file mode 100644 index 000000000..619ee9004 Binary files /dev/null and b/tests/data/aloha_sim_insertion_scripted/meta_data/stats.safetensors differ diff --git a/tests/data/aloha_sim_insertion_scripted/train/data-00000-of-00001.arrow b/tests/data/aloha_sim_insertion_scripted/train/data-00000-of-00001.arrow index 034f759f5..f5cdadcf5 100644 Binary files a/tests/data/aloha_sim_insertion_scripted/train/data-00000-of-00001.arrow and b/tests/data/aloha_sim_insertion_scripted/train/data-00000-of-00001.arrow differ diff --git a/tests/data/aloha_sim_insertion_scripted/train/dataset_info.json b/tests/data/aloha_sim_insertion_scripted/train/dataset_info.json index 542c7bf12..69f80830c 100644 --- a/tests/data/aloha_sim_insertion_scripted/train/dataset_info.json +++ b/tests/data/aloha_sim_insertion_scripted/train/dataset_info.json @@ -21,11 +21,11 @@ "length": 14, "_type": "Sequence" }, - "episode_id": { + "episode_index": { "dtype": "int64", "_type": "Value" }, - "frame_id": { + "frame_index": { "dtype": "int64", "_type": "Value" }, @@ -37,14 +37,6 @@ "dtype": "bool", "_type": "Value" }, - "episode_data_index_from": { - "dtype": "int64", - "_type": "Value" - }, - "episode_data_index_to": { - "dtype": "int64", - "_type": "Value" - }, "index": { "dtype": "int64", "_type": "Value" diff --git a/tests/data/aloha_sim_insertion_scripted/train/state.json b/tests/data/aloha_sim_insertion_scripted/train/state.json index ecaa8fd87..716aca60b 100644 --- a/tests/data/aloha_sim_insertion_scripted/train/state.json +++ b/tests/data/aloha_sim_insertion_scripted/train/state.json @@ -4,7 +4,7 @@ "filename": "data-00000-of-00001.arrow" } ], - "_fingerprint": "d8e4a817b5449498", + "_fingerprint": "97c28d4ad1536e4c", "_format_columns": null, "_format_kwargs": {}, "_format_type": "torch", diff --git a/tests/data/aloha_sim_transfer_cube_human/meta_data/episode_data_index.safetensors b/tests/data/aloha_sim_transfer_cube_human/meta_data/episode_data_index.safetensors new file mode 100644 index 000000000..8685de718 Binary files /dev/null and b/tests/data/aloha_sim_transfer_cube_human/meta_data/episode_data_index.safetensors differ diff --git a/tests/data/aloha_sim_transfer_cube_human/meta_data/info.json b/tests/data/aloha_sim_transfer_cube_human/meta_data/info.json new file mode 100644 index 000000000..02e62b67d --- /dev/null +++ b/tests/data/aloha_sim_transfer_cube_human/meta_data/info.json @@ -0,0 +1,3 @@ +{ + "fps": 50 +} \ No newline at end of file diff --git a/tests/data/aloha_sim_transfer_cube_human/meta_data/stats.safetensors b/tests/data/aloha_sim_transfer_cube_human/meta_data/stats.safetensors new file mode 100644 index 000000000..998e6105b Binary files /dev/null and b/tests/data/aloha_sim_transfer_cube_human/meta_data/stats.safetensors differ diff --git a/tests/data/aloha_sim_transfer_cube_human/train/data-00000-of-00001.arrow b/tests/data/aloha_sim_transfer_cube_human/train/data-00000-of-00001.arrow index 9682f0059..1bb1f5195 100644 Binary files a/tests/data/aloha_sim_transfer_cube_human/train/data-00000-of-00001.arrow and b/tests/data/aloha_sim_transfer_cube_human/train/data-00000-of-00001.arrow differ diff --git a/tests/data/aloha_sim_transfer_cube_human/train/dataset_info.json b/tests/data/aloha_sim_transfer_cube_human/train/dataset_info.json index 542c7bf12..69f80830c 100644 --- a/tests/data/aloha_sim_transfer_cube_human/train/dataset_info.json +++ b/tests/data/aloha_sim_transfer_cube_human/train/dataset_info.json @@ -21,11 +21,11 @@ "length": 14, "_type": "Sequence" }, - "episode_id": { + "episode_index": { "dtype": "int64", "_type": "Value" }, - "frame_id": { + "frame_index": { "dtype": "int64", "_type": "Value" }, @@ -37,14 +37,6 @@ "dtype": "bool", "_type": "Value" }, - "episode_data_index_from": { - "dtype": "int64", - "_type": "Value" - }, - "episode_data_index_to": { - "dtype": "int64", - "_type": "Value" - }, "index": { "dtype": "int64", "_type": "Value" diff --git a/tests/data/aloha_sim_transfer_cube_human/train/state.json b/tests/data/aloha_sim_transfer_cube_human/train/state.json index 0167986ba..d9449a3c4 100644 --- a/tests/data/aloha_sim_transfer_cube_human/train/state.json +++ b/tests/data/aloha_sim_transfer_cube_human/train/state.json @@ -4,7 +4,7 @@ "filename": "data-00000-of-00001.arrow" } ], - "_fingerprint": "f03482befa767127", + "_fingerprint": "cb9349b5c92951e8", "_format_columns": null, "_format_kwargs": {}, "_format_type": "torch", diff --git a/tests/data/aloha_sim_transfer_cube_scripted/meta_data/episode_data_index.safetensors b/tests/data/aloha_sim_transfer_cube_scripted/meta_data/episode_data_index.safetensors new file mode 100644 index 000000000..8685de718 Binary files /dev/null and b/tests/data/aloha_sim_transfer_cube_scripted/meta_data/episode_data_index.safetensors differ diff --git a/tests/data/aloha_sim_transfer_cube_scripted/meta_data/info.json b/tests/data/aloha_sim_transfer_cube_scripted/meta_data/info.json new file mode 100644 index 000000000..02e62b67d --- /dev/null +++ b/tests/data/aloha_sim_transfer_cube_scripted/meta_data/info.json @@ -0,0 +1,3 @@ +{ + "fps": 50 +} \ No newline at end of file diff --git a/tests/data/aloha_sim_transfer_cube_scripted/meta_data/stats.safetensors b/tests/data/aloha_sim_transfer_cube_scripted/meta_data/stats.safetensors new file mode 100644 index 000000000..91696d32d Binary files /dev/null and b/tests/data/aloha_sim_transfer_cube_scripted/meta_data/stats.safetensors differ diff --git a/tests/data/aloha_sim_transfer_cube_scripted/train/data-00000-of-00001.arrow b/tests/data/aloha_sim_transfer_cube_scripted/train/data-00000-of-00001.arrow index 567191d54..d658a6dfe 100644 Binary files a/tests/data/aloha_sim_transfer_cube_scripted/train/data-00000-of-00001.arrow and b/tests/data/aloha_sim_transfer_cube_scripted/train/data-00000-of-00001.arrow differ diff --git a/tests/data/aloha_sim_transfer_cube_scripted/train/dataset_info.json b/tests/data/aloha_sim_transfer_cube_scripted/train/dataset_info.json index 542c7bf12..69f80830c 100644 --- a/tests/data/aloha_sim_transfer_cube_scripted/train/dataset_info.json +++ b/tests/data/aloha_sim_transfer_cube_scripted/train/dataset_info.json @@ -21,11 +21,11 @@ "length": 14, "_type": "Sequence" }, - "episode_id": { + "episode_index": { "dtype": "int64", "_type": "Value" }, - "frame_id": { + "frame_index": { "dtype": "int64", "_type": "Value" }, @@ -37,14 +37,6 @@ "dtype": "bool", "_type": "Value" }, - "episode_data_index_from": { - "dtype": "int64", - "_type": "Value" - }, - "episode_data_index_to": { - "dtype": "int64", - "_type": "Value" - }, "index": { "dtype": "int64", "_type": "Value" diff --git a/tests/data/aloha_sim_transfer_cube_scripted/train/state.json b/tests/data/aloha_sim_transfer_cube_scripted/train/state.json index 56005bc91..2d4dfc64f 100644 --- a/tests/data/aloha_sim_transfer_cube_scripted/train/state.json +++ b/tests/data/aloha_sim_transfer_cube_scripted/train/state.json @@ -4,7 +4,7 @@ "filename": "data-00000-of-00001.arrow" } ], - "_fingerprint": "93e03c6320c7d56e", + "_fingerprint": "e4d7ad2b360db1af", "_format_columns": null, "_format_kwargs": {}, "_format_type": "torch", diff --git a/tests/data/pusht/meta_data/episode_data_index.safetensors b/tests/data/pusht/meta_data/episode_data_index.safetensors new file mode 100644 index 000000000..9343d2d56 Binary files /dev/null and b/tests/data/pusht/meta_data/episode_data_index.safetensors differ diff --git a/tests/data/pusht/meta_data/info.json b/tests/data/pusht/meta_data/info.json new file mode 100644 index 000000000..5c9a8aef0 --- /dev/null +++ b/tests/data/pusht/meta_data/info.json @@ -0,0 +1,3 @@ +{ + "fps": 10 +} \ No newline at end of file diff --git a/tests/data/pusht/meta_data/stats.safetensors b/tests/data/pusht/meta_data/stats.safetensors new file mode 100644 index 000000000..fa2380e6a Binary files /dev/null and b/tests/data/pusht/meta_data/stats.safetensors differ diff --git a/tests/data/pusht/train/data-00000-of-00001.arrow b/tests/data/pusht/train/data-00000-of-00001.arrow index 9a36a8db0..5972be9bf 100644 Binary files a/tests/data/pusht/train/data-00000-of-00001.arrow and b/tests/data/pusht/train/data-00000-of-00001.arrow differ diff --git a/tests/data/pusht/train/dataset_info.json b/tests/data/pusht/train/dataset_info.json index 667e06f73..aefe47858 100644 --- a/tests/data/pusht/train/dataset_info.json +++ b/tests/data/pusht/train/dataset_info.json @@ -21,11 +21,11 @@ "length": 2, "_type": "Sequence" }, - "episode_id": { + "episode_index": { "dtype": "int64", "_type": "Value" }, - "frame_id": { + "frame_index": { "dtype": "int64", "_type": "Value" }, @@ -45,14 +45,6 @@ "dtype": "bool", "_type": "Value" }, - "episode_data_index_from": { - "dtype": "int64", - "_type": "Value" - }, - "episode_data_index_to": { - "dtype": "int64", - "_type": "Value" - }, "index": { "dtype": "int64", "_type": "Value" diff --git a/tests/data/pusht/train/meta_data/episode_data_index.safetensors b/tests/data/pusht/train/meta_data/episode_data_index.safetensors new file mode 100644 index 000000000..3511c2668 Binary files /dev/null and b/tests/data/pusht/train/meta_data/episode_data_index.safetensors differ diff --git a/tests/data/pusht/train/meta_data/info.json b/tests/data/pusht/train/meta_data/info.json new file mode 100644 index 000000000..5c9a8aef0 --- /dev/null +++ b/tests/data/pusht/train/meta_data/info.json @@ -0,0 +1,3 @@ +{ + "fps": 10 +} \ No newline at end of file diff --git a/tests/data/pusht/train/meta_data/stats_action.safetensors b/tests/data/pusht/train/meta_data/stats_action.safetensors new file mode 100644 index 000000000..2c2553b1d Binary files /dev/null and b/tests/data/pusht/train/meta_data/stats_action.safetensors differ diff --git a/tests/data/pusht/train/meta_data/stats_observation.image.safetensors b/tests/data/pusht/train/meta_data/stats_observation.image.safetensors new file mode 100644 index 000000000..0a145d477 Binary files /dev/null and b/tests/data/pusht/train/meta_data/stats_observation.image.safetensors differ diff --git a/tests/data/pusht/train/meta_data/stats_observation.state.safetensors b/tests/data/pusht/train/meta_data/stats_observation.state.safetensors new file mode 100644 index 000000000..28ee2851d Binary files /dev/null and b/tests/data/pusht/train/meta_data/stats_observation.state.safetensors differ diff --git a/tests/data/pusht/train/state.json b/tests/data/pusht/train/state.json index 7e0ff5742..dda3f8897 100644 --- a/tests/data/pusht/train/state.json +++ b/tests/data/pusht/train/state.json @@ -4,7 +4,7 @@ "filename": "data-00000-of-00001.arrow" } ], - "_fingerprint": "21bb9a76ed78a475", + "_fingerprint": "a04a9ce660122e23", "_format_columns": null, "_format_kwargs": {}, "_format_type": "torch", diff --git a/tests/data/xarm_lift_medium/meta_data/episode_data_index.safetensors b/tests/data/xarm_lift_medium/meta_data/episode_data_index.safetensors new file mode 100644 index 000000000..721609343 Binary files /dev/null and b/tests/data/xarm_lift_medium/meta_data/episode_data_index.safetensors differ diff --git a/tests/data/xarm_lift_medium/meta_data/info.json b/tests/data/xarm_lift_medium/meta_data/info.json new file mode 100644 index 000000000..f9d6b3032 --- /dev/null +++ b/tests/data/xarm_lift_medium/meta_data/info.json @@ -0,0 +1,3 @@ +{ + "fps": 15 +} \ No newline at end of file diff --git a/tests/data/xarm_lift_medium/meta_data/stats.safetensors b/tests/data/xarm_lift_medium/meta_data/stats.safetensors new file mode 100644 index 000000000..bdcc1b0fd Binary files /dev/null and b/tests/data/xarm_lift_medium/meta_data/stats.safetensors differ diff --git a/tests/data/xarm_lift_medium/train/data-00000-of-00001.arrow b/tests/data/xarm_lift_medium/train/data-00000-of-00001.arrow index 45d527e0f..d62121010 100644 Binary files a/tests/data/xarm_lift_medium/train/data-00000-of-00001.arrow and b/tests/data/xarm_lift_medium/train/data-00000-of-00001.arrow differ diff --git a/tests/data/xarm_lift_medium/train/dataset_info.json b/tests/data/xarm_lift_medium/train/dataset_info.json index bb647c412..59a43bdea 100644 --- a/tests/data/xarm_lift_medium/train/dataset_info.json +++ b/tests/data/xarm_lift_medium/train/dataset_info.json @@ -21,11 +21,11 @@ "length": 4, "_type": "Sequence" }, - "episode_id": { + "episode_index": { "dtype": "int64", "_type": "Value" }, - "frame_id": { + "frame_index": { "dtype": "int64", "_type": "Value" }, @@ -41,14 +41,6 @@ "dtype": "bool", "_type": "Value" }, - "episode_data_index_from": { - "dtype": "int64", - "_type": "Value" - }, - "episode_data_index_to": { - "dtype": "int64", - "_type": "Value" - }, "index": { "dtype": "int64", "_type": "Value" diff --git a/tests/data/xarm_lift_medium/train/state.json b/tests/data/xarm_lift_medium/train/state.json index c930c52ce..642fda30a 100644 --- a/tests/data/xarm_lift_medium/train/state.json +++ b/tests/data/xarm_lift_medium/train/state.json @@ -4,7 +4,7 @@ "filename": "data-00000-of-00001.arrow" } ], - "_fingerprint": "a95cbec45e3bb9d6", + "_fingerprint": "cc6afdfcdd6f63ab", "_format_columns": null, "_format_kwargs": {}, "_format_type": "torch", diff --git a/tests/data/xarm_lift_medium_replay/meta_data/episode_data_index.safetensors b/tests/data/xarm_lift_medium_replay/meta_data/episode_data_index.safetensors new file mode 100644 index 000000000..721609343 Binary files /dev/null and b/tests/data/xarm_lift_medium_replay/meta_data/episode_data_index.safetensors differ diff --git a/tests/data/xarm_lift_medium_replay/meta_data/info.json b/tests/data/xarm_lift_medium_replay/meta_data/info.json new file mode 100644 index 000000000..f9d6b3032 --- /dev/null +++ b/tests/data/xarm_lift_medium_replay/meta_data/info.json @@ -0,0 +1,3 @@ +{ + "fps": 15 +} \ No newline at end of file diff --git a/tests/data/xarm_lift_medium_replay/meta_data/stats.safetensors b/tests/data/xarm_lift_medium_replay/meta_data/stats.safetensors new file mode 100644 index 000000000..4808895eb Binary files /dev/null and b/tests/data/xarm_lift_medium_replay/meta_data/stats.safetensors differ diff --git a/tests/data/xarm_lift_medium_replay/train/data-00000-of-00001.arrow b/tests/data/xarm_lift_medium_replay/train/data-00000-of-00001.arrow new file mode 100644 index 000000000..b52481129 Binary files /dev/null and b/tests/data/xarm_lift_medium_replay/train/data-00000-of-00001.arrow differ diff --git a/tests/data/xarm_lift_medium_replay/train/dataset_info.json b/tests/data/xarm_lift_medium_replay/train/dataset_info.json new file mode 100644 index 000000000..59a43bdea --- /dev/null +++ b/tests/data/xarm_lift_medium_replay/train/dataset_info.json @@ -0,0 +1,51 @@ +{ + "citation": "", + "description": "", + "features": { + "observation.image": { + "_type": "Image" + }, + "observation.state": { + "feature": { + "dtype": "float32", + "_type": "Value" + }, + "length": 4, + "_type": "Sequence" + }, + "action": { + "feature": { + "dtype": "float32", + "_type": "Value" + }, + "length": 4, + "_type": "Sequence" + }, + "episode_index": { + "dtype": "int64", + "_type": "Value" + }, + "frame_index": { + "dtype": "int64", + "_type": "Value" + }, + "timestamp": { + "dtype": "float32", + "_type": "Value" + }, + "next.reward": { + "dtype": "float32", + "_type": "Value" + }, + "next.done": { + "dtype": "bool", + "_type": "Value" + }, + "index": { + "dtype": "int64", + "_type": "Value" + } + }, + "homepage": "", + "license": "" +} \ No newline at end of file diff --git a/tests/data/xarm_lift_medium_replay/train/state.json b/tests/data/xarm_lift_medium_replay/train/state.json new file mode 100644 index 000000000..e9b74d7de --- /dev/null +++ b/tests/data/xarm_lift_medium_replay/train/state.json @@ -0,0 +1,13 @@ +{ + "_data_files": [ + { + "filename": "data-00000-of-00001.arrow" + } + ], + "_fingerprint": "9f8e1a8c1845df55", + "_format_columns": null, + "_format_kwargs": {}, + "_format_type": "torch", + "_output_all_columns": false, + "_split": null +} \ No newline at end of file diff --git a/tests/data/xarm_push_medium/meta_data/episode_data_index.safetensors b/tests/data/xarm_push_medium/meta_data/episode_data_index.safetensors new file mode 100644 index 000000000..721609343 Binary files /dev/null and b/tests/data/xarm_push_medium/meta_data/episode_data_index.safetensors differ diff --git a/tests/data/xarm_push_medium/meta_data/info.json b/tests/data/xarm_push_medium/meta_data/info.json new file mode 100644 index 000000000..f9d6b3032 --- /dev/null +++ b/tests/data/xarm_push_medium/meta_data/info.json @@ -0,0 +1,3 @@ +{ + "fps": 15 +} \ No newline at end of file diff --git a/tests/data/xarm_push_medium/meta_data/stats.safetensors b/tests/data/xarm_push_medium/meta_data/stats.safetensors new file mode 100644 index 000000000..f216e0587 Binary files /dev/null and b/tests/data/xarm_push_medium/meta_data/stats.safetensors differ diff --git a/tests/data/xarm_push_medium/train/data-00000-of-00001.arrow b/tests/data/xarm_push_medium/train/data-00000-of-00001.arrow new file mode 100644 index 000000000..241117c58 Binary files /dev/null and b/tests/data/xarm_push_medium/train/data-00000-of-00001.arrow differ diff --git a/tests/data/xarm_push_medium/train/dataset_info.json b/tests/data/xarm_push_medium/train/dataset_info.json new file mode 100644 index 000000000..9e47b3456 --- /dev/null +++ b/tests/data/xarm_push_medium/train/dataset_info.json @@ -0,0 +1,51 @@ +{ + "citation": "", + "description": "", + "features": { + "observation.image": { + "_type": "Image" + }, + "observation.state": { + "feature": { + "dtype": "float32", + "_type": "Value" + }, + "length": 4, + "_type": "Sequence" + }, + "action": { + "feature": { + "dtype": "float32", + "_type": "Value" + }, + "length": 3, + "_type": "Sequence" + }, + "episode_index": { + "dtype": "int64", + "_type": "Value" + }, + "frame_index": { + "dtype": "int64", + "_type": "Value" + }, + "timestamp": { + "dtype": "float32", + "_type": "Value" + }, + "next.reward": { + "dtype": "float32", + "_type": "Value" + }, + "next.done": { + "dtype": "bool", + "_type": "Value" + }, + "index": { + "dtype": "int64", + "_type": "Value" + } + }, + "homepage": "", + "license": "" +} \ No newline at end of file diff --git a/tests/data/xarm_push_medium/train/state.json b/tests/data/xarm_push_medium/train/state.json new file mode 100644 index 000000000..0ec1f0465 --- /dev/null +++ b/tests/data/xarm_push_medium/train/state.json @@ -0,0 +1,13 @@ +{ + "_data_files": [ + { + "filename": "data-00000-of-00001.arrow" + } + ], + "_fingerprint": "c900258061dd0b3f", + "_format_columns": null, + "_format_kwargs": {}, + "_format_type": "torch", + "_output_all_columns": false, + "_split": null +} \ No newline at end of file diff --git a/tests/data/xarm_push_medium_replay/meta_data/episode_data_index.safetensors b/tests/data/xarm_push_medium_replay/meta_data/episode_data_index.safetensors new file mode 100644 index 000000000..721609343 Binary files /dev/null and b/tests/data/xarm_push_medium_replay/meta_data/episode_data_index.safetensors differ diff --git a/tests/data/xarm_push_medium_replay/meta_data/info.json b/tests/data/xarm_push_medium_replay/meta_data/info.json new file mode 100644 index 000000000..f9d6b3032 --- /dev/null +++ b/tests/data/xarm_push_medium_replay/meta_data/info.json @@ -0,0 +1,3 @@ +{ + "fps": 15 +} \ No newline at end of file diff --git a/tests/data/xarm_push_medium_replay/meta_data/stats.safetensors b/tests/data/xarm_push_medium_replay/meta_data/stats.safetensors new file mode 100644 index 000000000..0de47551e Binary files /dev/null and b/tests/data/xarm_push_medium_replay/meta_data/stats.safetensors differ diff --git a/tests/data/xarm_push_medium_replay/train/data-00000-of-00001.arrow b/tests/data/xarm_push_medium_replay/train/data-00000-of-00001.arrow new file mode 100644 index 000000000..2e07ea95f Binary files /dev/null and b/tests/data/xarm_push_medium_replay/train/data-00000-of-00001.arrow differ diff --git a/tests/data/xarm_push_medium_replay/train/dataset_info.json b/tests/data/xarm_push_medium_replay/train/dataset_info.json new file mode 100644 index 000000000..9e47b3456 --- /dev/null +++ b/tests/data/xarm_push_medium_replay/train/dataset_info.json @@ -0,0 +1,51 @@ +{ + "citation": "", + "description": "", + "features": { + "observation.image": { + "_type": "Image" + }, + "observation.state": { + "feature": { + "dtype": "float32", + "_type": "Value" + }, + "length": 4, + "_type": "Sequence" + }, + "action": { + "feature": { + "dtype": "float32", + "_type": "Value" + }, + "length": 3, + "_type": "Sequence" + }, + "episode_index": { + "dtype": "int64", + "_type": "Value" + }, + "frame_index": { + "dtype": "int64", + "_type": "Value" + }, + "timestamp": { + "dtype": "float32", + "_type": "Value" + }, + "next.reward": { + "dtype": "float32", + "_type": "Value" + }, + "next.done": { + "dtype": "bool", + "_type": "Value" + }, + "index": { + "dtype": "int64", + "_type": "Value" + } + }, + "homepage": "", + "license": "" +} \ No newline at end of file diff --git a/tests/data/xarm_push_medium_replay/train/state.json b/tests/data/xarm_push_medium_replay/train/state.json new file mode 100644 index 000000000..39ffeaf03 --- /dev/null +++ b/tests/data/xarm_push_medium_replay/train/state.json @@ -0,0 +1,13 @@ +{ + "_data_files": [ + { + "filename": "data-00000-of-00001.arrow" + } + ], + "_fingerprint": "e51c80a33c7688c0", + "_format_columns": null, + "_format_kwargs": {}, + "_format_type": "torch", + "_output_all_columns": false, + "_split": null +} \ No newline at end of file