wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Add tests/data

This commit is contained in:
Cadene 2024-04-19 23:19:16 +00:00
parent c20cf2fbbc
commit 20928021c0
59 changed files with 240 additions and 66 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

BIN
tests/data/aloha_sim_insertion_human/meta_data/episode_data_index.safetensors Normal file
View File

Binary file not shown.

3
tests/data/aloha_sim_insertion_human/meta_data/info.json Normal file
View File

@ -0,0 +1,3 @@
{
"fps": 50
}

BIN
tests/data/aloha_sim_insertion_human/meta_data/stats.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/aloha_sim_insertion_human/train/data-00000-of-00001.arrow
View File

Binary file not shown.

12
tests/data/aloha_sim_insertion_human/train/dataset_info.json
View File

@ -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"

2
tests/data/aloha_sim_insertion_human/train/state.json
View File

@ -4,7 +4,7 @@
"filename": "data-00000-of-00001.arrow"
}
],
"_fingerprint": "d79cf82ffc86f110",
"_fingerprint": "22eeca7a3f4725ee",
"_format_columns": null,
"_format_kwargs": {},
"_format_type": "torch",

BIN
tests/data/aloha_sim_insertion_scripted/meta_data/episode_data_index.safetensors Normal file
View File

Binary file not shown.

3
tests/data/aloha_sim_insertion_scripted/meta_data/info.json Normal file
View File

@ -0,0 +1,3 @@
{
"fps": 50
}

BIN
tests/data/aloha_sim_insertion_scripted/meta_data/stats.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/aloha_sim_insertion_scripted/train/data-00000-of-00001.arrow
View File

Binary file not shown.

12
tests/data/aloha_sim_insertion_scripted/train/dataset_info.json
View File

@ -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"

2
tests/data/aloha_sim_insertion_scripted/train/state.json
View File

@ -4,7 +4,7 @@
"filename": "data-00000-of-00001.arrow"
}
],
"_fingerprint": "d8e4a817b5449498",
"_fingerprint": "97c28d4ad1536e4c",
"_format_columns": null,
"_format_kwargs": {},
"_format_type": "torch",

BIN
tests/data/aloha_sim_transfer_cube_human/meta_data/episode_data_index.safetensors Normal file
View File

Binary file not shown.

3
tests/data/aloha_sim_transfer_cube_human/meta_data/info.json Normal file
View File

@ -0,0 +1,3 @@
{
"fps": 50
}

BIN
tests/data/aloha_sim_transfer_cube_human/meta_data/stats.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/aloha_sim_transfer_cube_human/train/data-00000-of-00001.arrow
View File

Binary file not shown.

12
tests/data/aloha_sim_transfer_cube_human/train/dataset_info.json
View File

@ -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"

2
tests/data/aloha_sim_transfer_cube_human/train/state.json
View File

@ -4,7 +4,7 @@
"filename": "data-00000-of-00001.arrow"
}
],
"_fingerprint": "f03482befa767127",
"_fingerprint": "cb9349b5c92951e8",
"_format_columns": null,
"_format_kwargs": {},
"_format_type": "torch",

BIN
tests/data/aloha_sim_transfer_cube_scripted/meta_data/episode_data_index.safetensors Normal file
View File

Binary file not shown.

3
tests/data/aloha_sim_transfer_cube_scripted/meta_data/info.json Normal file
View File

@ -0,0 +1,3 @@
{
"fps": 50
}

BIN
tests/data/aloha_sim_transfer_cube_scripted/meta_data/stats.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/aloha_sim_transfer_cube_scripted/train/data-00000-of-00001.arrow
View File

Binary file not shown.

12
tests/data/aloha_sim_transfer_cube_scripted/train/dataset_info.json
View File

@ -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"

2
tests/data/aloha_sim_transfer_cube_scripted/train/state.json
View File

@ -4,7 +4,7 @@
"filename": "data-00000-of-00001.arrow"
}
],
"_fingerprint": "93e03c6320c7d56e",
"_fingerprint": "e4d7ad2b360db1af",
"_format_columns": null,
"_format_kwargs": {},
"_format_type": "torch",

BIN
tests/data/pusht/meta_data/episode_data_index.safetensors Normal file
View File

Binary file not shown.

3
tests/data/pusht/meta_data/info.json Normal file
View File

@ -0,0 +1,3 @@
{
"fps": 10
}

BIN
tests/data/pusht/meta_data/stats.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/pusht/train/data-00000-of-00001.arrow
View File

Binary file not shown.

12
tests/data/pusht/train/dataset_info.json
View File

@ -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"

BIN
tests/data/pusht/train/meta_data/episode_data_index.safetensors Normal file
View File

Binary file not shown.

3
tests/data/pusht/train/meta_data/info.json Normal file
View File

@ -0,0 +1,3 @@
{
"fps": 10
}

BIN
tests/data/pusht/train/meta_data/stats_action.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/pusht/train/meta_data/stats_observation.image.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/pusht/train/meta_data/stats_observation.state.safetensors Normal file
View File

Binary file not shown.

2
tests/data/pusht/train/state.json
View File

@ -4,7 +4,7 @@
"filename": "data-00000-of-00001.arrow"
}
],
"_fingerprint": "21bb9a76ed78a475",
"_fingerprint": "a04a9ce660122e23",
"_format_columns": null,
"_format_kwargs": {},
"_format_type": "torch",

BIN
tests/data/xarm_lift_medium/meta_data/episode_data_index.safetensors Normal file
View File

Binary file not shown.

3
tests/data/xarm_lift_medium/meta_data/info.json Normal file
View File

@ -0,0 +1,3 @@
{
"fps": 15
}

BIN
tests/data/xarm_lift_medium/meta_data/stats.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/xarm_lift_medium/train/data-00000-of-00001.arrow
View File

Binary file not shown.

12
tests/data/xarm_lift_medium/train/dataset_info.json
View File

@ -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"

2
tests/data/xarm_lift_medium/train/state.json
View File

@ -4,7 +4,7 @@
"filename": "data-00000-of-00001.arrow"
}
],
"_fingerprint": "a95cbec45e3bb9d6",
"_fingerprint": "cc6afdfcdd6f63ab",
"_format_columns": null,
"_format_kwargs": {},
"_format_type": "torch",

BIN
tests/data/xarm_lift_medium_replay/meta_data/episode_data_index.safetensors Normal file
View File

Binary file not shown.

3
tests/data/xarm_lift_medium_replay/meta_data/info.json Normal file
View File

@ -0,0 +1,3 @@
{
"fps": 15
}

BIN
tests/data/xarm_lift_medium_replay/meta_data/stats.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/xarm_lift_medium_replay/train/data-00000-of-00001.arrow Normal file
View File

Binary file not shown.

51
tests/data/xarm_lift_medium_replay/train/dataset_info.json Normal file
View File

@ -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": ""
}

13
tests/data/xarm_lift_medium_replay/train/state.json Normal file
View File

@ -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
}

BIN
tests/data/xarm_push_medium/meta_data/episode_data_index.safetensors Normal file
View File

Binary file not shown.

3
tests/data/xarm_push_medium/meta_data/info.json Normal file
View File

@ -0,0 +1,3 @@
{
"fps": 15
}

BIN
tests/data/xarm_push_medium/meta_data/stats.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/xarm_push_medium/train/data-00000-of-00001.arrow Normal file
View File

Binary file not shown.

51
tests/data/xarm_push_medium/train/dataset_info.json Normal file
View File

@ -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": ""
}

13
tests/data/xarm_push_medium/train/state.json Normal file
View File

@ -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
}

BIN
tests/data/xarm_push_medium_replay/meta_data/episode_data_index.safetensors Normal file
View File

Binary file not shown.

3
tests/data/xarm_push_medium_replay/meta_data/info.json Normal file
View File

@ -0,0 +1,3 @@
{
"fps": 15
}

BIN
tests/data/xarm_push_medium_replay/meta_data/stats.safetensors Normal file
View File

Binary file not shown.

BIN
tests/data/xarm_push_medium_replay/train/data-00000-of-00001.arrow Normal file
View File

Binary file not shown.

51
tests/data/xarm_push_medium_replay/train/dataset_info.json Normal file
View File

@ -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": ""
}

13
tests/data/xarm_push_medium_replay/train/state.json Normal file
View File

@ -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
}