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/pusht/train/data-00000-of-00001.arrow
+++ b/tests/data/pusht/train/data-00000-of-00001.arrow
--- a/tests/data/pusht/train/dataset_info.json
+++ b/tests/data/pusht/train/dataset_info.json
@@ -45,11 +45,11 @@
      "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"
    },
--- 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": "f7ed966ae18000ae",
+  "_fingerprint": "21bb9a76ed78a475",
  "_format_columns": null,
  "_format_kwargs": {},
  "_format_type": "torch",