Address comments

2024-04-16 17:14:40 +00:00
parent b241ea46dd
commit 36d9e885ef
24 changed files with 100 additions and 94 deletions
--- a/tests/data/aloha_sim_insertion_scripted/train/dataset_info.json
+++ b/tests/data/aloha_sim_insertion_scripted/train/dataset_info.json
@@ -2,6 +2,9 @@
  "citation": "",
  "description": "",
  "features": {
+    "observation.images.top": {
+      "_type": "Image"
+    },
    "observation.state": {
      "feature": {
        "dtype": "float32",
@@ -34,17 +37,14 @@
      "dtype": "bool",
      "_type": "Value"
    },
-    "episode_data_id_from": {
+    "episode_data_index_from": {
      "dtype": "int64",
      "_type": "Value"
    },
-    "episode_data_id_to": {
+    "episode_data_index_to": {
      "dtype": "int64",
      "_type": "Value"
    },
-    "observation.images.top": {
-      "_type": "Image"
-    },
    "index": {
      "dtype": "int64",
      "_type": "Value"