test_datasets.py are passing!

2026-06-01 03:11:29 +00:00 · 2024-04-08 14:02:03 +00:00
parent e1ac5dc62f
commit 70aaf1c4cb
109 changed files with 90 additions and 228 deletions
--- a/tests/data/aloha_sim_insertion_scripted/data_dict.pth
+++ b/tests/data/aloha_sim_insertion_scripted/data_dict.pth
--- a/tests/data/aloha_sim_insertion_scripted/data_ids_per_episode.pth
+++ b/tests/data/aloha_sim_insertion_scripted/data_ids_per_episode.pth
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/action.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/action.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/episode.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/episode.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/frame_id.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/frame_id.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/meta.json
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/meta.json
@@ -1 +0,0 @@
-{"action": {"device": "cpu", "shape": [50, 14], "dtype": "torch.float32"}, "episode": {"device": "cpu", "shape": [50], "dtype": "torch.int64"}, "frame_id": {"device": "cpu", "shape": [50], "dtype": "torch.int64"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/done.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/done.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/meta.json
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/meta.json
@@ -1 +0,0 @@
-{"done": {"device": "cpu", "shape": [50, 1], "dtype": "torch.bool"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/image/meta.json
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/image/meta.json
@@ -1 +0,0 @@
-{"top": {"device": "cpu", "shape": [50, 3, 480, 640], "dtype": "torch.uint8"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/image/top.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/image/top.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/meta.json
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/meta.json
@@ -1 +0,0 @@
-{"state": {"device": "cpu", "shape": [50, 14], "dtype": "torch.float32"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/state.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/next/observation/state.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/image/meta.json
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/image/meta.json
@@ -1 +0,0 @@
-{"top": {"device": "cpu", "shape": [50, 3, 480, 640], "dtype": "torch.uint8"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/image/top.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/image/top.memmap
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/meta.json
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/meta.json
@@ -1 +0,0 @@
-{"state": {"device": "cpu", "shape": [50, 14], "dtype": "torch.float32"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}
--- a/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/state.memmap
+++ b/tests/data/aloha_sim_insertion_scripted/replay_buffer/observation/state.memmap
--- a/tests/data/aloha_sim_insertion_scripted/stats.pth
+++ b/tests/data/aloha_sim_insertion_scripted/stats.pth
				`@@ -1 +0,0 @@`
				`{"action": {"device": "cpu", "shape": [50, 14], "dtype": "torch.float32"}, "episode": {"device": "cpu", "shape": [50], "dtype": "torch.int64"}, "frame_id": {"device": "cpu", "shape": [50], "dtype": "torch.int64"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}`
				`@@ -1 +0,0 @@`
				`{"done": {"device": "cpu", "shape": [50, 1], "dtype": "torch.bool"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}`
				`@@ -1 +0,0 @@`
				`{"top": {"device": "cpu", "shape": [50, 3, 480, 640], "dtype": "torch.uint8"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}`
				`@@ -1 +0,0 @@`
				`{"state": {"device": "cpu", "shape": [50, 14], "dtype": "torch.float32"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}`