Address comments

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"

tests/data/aloha_sim_insertion_scripted/train/state.json

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