Add tests/data

2024-04-19 23:19:16 +00:00
parent c20cf2fbbc
commit 20928021c0
59 changed files with 240 additions and 66 deletions
--- 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
--- 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"
--- 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",