diff --git a/tests/data/pusht/action.memmap b/tests/data/pusht/action.memmap
new file mode 100644
index 00000000..f4127fb1
--- /dev/null
+++ b/tests/data/pusht/action.memmap
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ba17d8e5c30151ea5f7f6fc31f19e12a68ce2113774b74c8aca0c7ef962a75f4
+size 400
diff --git a/tests/data/pusht/episode.memmap b/tests/data/pusht/episode.memmap
new file mode 100644
index 00000000..af9fb07f
--- /dev/null
+++ b/tests/data/pusht/episode.memmap
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:7a12e561363385e9dfeeab326368731c030ed4b374e7f5897ac819159d2884c5
+size 400
diff --git a/tests/data/pusht/frame_id.memmap b/tests/data/pusht/frame_id.memmap
new file mode 100644
index 00000000..dc2f585c
--- /dev/null
+++ b/tests/data/pusht/frame_id.memmap
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c202d9cfc7858fd49d522047e16948359bbbb2eda2d3825d552e45a78d5f8585
+size 400
diff --git a/tests/data/pusht/meta.json b/tests/data/pusht/meta.json
new file mode 100644
index 00000000..6f7c4218
--- /dev/null
+++ b/tests/data/pusht/meta.json
@@ -0,0 +1 @@
+{"action": {"device": "cpu", "shape": [50, 2], "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'>"}
\ No newline at end of file
diff --git a/tests/data/pusht/next/done.memmap b/tests/data/pusht/next/done.memmap
new file mode 100644
index 00000000..44fd709f
--- /dev/null
+++ b/tests/data/pusht/next/done.memmap
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cc2786e1f9910a9d811400edcddaf7075195f7a16b216dcbefba3bc7c4f2ae51
+size 50
diff --git a/tests/data/pusht/next/meta.json b/tests/data/pusht/next/meta.json
new file mode 100644
index 00000000..b29a9ff7
--- /dev/null
+++ b/tests/data/pusht/next/meta.json
@@ -0,0 +1 @@
+{"reward": {"device": "cpu", "shape": [50, 1], "dtype": "torch.float32"}, "done": {"device": "cpu", "shape": [50, 1], "dtype": "torch.bool"}, "success": {"device": "cpu", "shape": [50, 1], "dtype": "torch.bool"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}
\ No newline at end of file
diff --git a/tests/data/pusht/next/observation/image.memmap b/tests/data/pusht/next/observation/image.memmap
new file mode 100644
index 00000000..68634378
--- /dev/null
+++ b/tests/data/pusht/next/observation/image.memmap
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ff6a3748c8223a82e54c61442df7b8baf478a20497ee2353645a1e9ccd765162
+size 5529600
diff --git a/tests/data/pusht/next/observation/meta.json b/tests/data/pusht/next/observation/meta.json
new file mode 100644
index 00000000..57e0edea
--- /dev/null
+++ b/tests/data/pusht/next/observation/meta.json
@@ -0,0 +1 @@
+{"image": {"device": "cpu", "shape": [50, 3, 96, 96], "dtype": "torch.float32"}, "state": {"device": "cpu", "shape": [50, 2], "dtype": "torch.float32"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}
\ No newline at end of file
diff --git a/tests/data/pusht/next/observation/state.memmap b/tests/data/pusht/next/observation/state.memmap
new file mode 100644
index 00000000..8dd28f2a
--- /dev/null
+++ b/tests/data/pusht/next/observation/state.memmap
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:fad4ece6d5fd66bbafa34f6ff383c483410082b8d7d4f4616808c3c458ce1d43
+size 400
diff --git a/tests/data/pusht/next/reward.memmap b/tests/data/pusht/next/reward.memmap
new file mode 100644
index 00000000..109ed5ad
--- /dev/null
+++ b/tests/data/pusht/next/reward.memmap
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:6d9c54dee5660c46886f32d80e57e9dd0ffa57ee0cd2a762b036d9c8e0c3a33a
+size 200
diff --git a/tests/data/pusht/next/success.memmap b/tests/data/pusht/next/success.memmap
new file mode 100644
index 00000000..44fd709f
--- /dev/null
+++ b/tests/data/pusht/next/success.memmap
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:cc2786e1f9910a9d811400edcddaf7075195f7a16b216dcbefba3bc7c4f2ae51
+size 50
diff --git a/tests/data/pusht/observation/image.memmap b/tests/data/pusht/observation/image.memmap
new file mode 100644
index 00000000..42c86ef0
--- /dev/null
+++ b/tests/data/pusht/observation/image.memmap
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:4bbde5cfd8cff9fd9fc6c9a57177f6fd31c8a03cf853b7d2234312f38380b0ba
+size 5529600
diff --git a/tests/data/pusht/observation/meta.json b/tests/data/pusht/observation/meta.json
new file mode 100644
index 00000000..57e0edea
--- /dev/null
+++ b/tests/data/pusht/observation/meta.json
@@ -0,0 +1 @@
+{"image": {"device": "cpu", "shape": [50, 3, 96, 96], "dtype": "torch.float32"}, "state": {"device": "cpu", "shape": [50, 2], "dtype": "torch.float32"}, "shape": [50], "device": "cpu", "_type": "<class 'tensordict._td.TensorDict'>"}
\ No newline at end of file
diff --git a/tests/data/pusht/observation/state.memmap b/tests/data/pusht/observation/state.memmap
new file mode 100644
index 00000000..3ac8e4ab
--- /dev/null
+++ b/tests/data/pusht/observation/state.memmap
@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:67c7e39090a16546fb1eade833d704f26464d574d7e431415f828159a154d2bf
+size 400
diff --git a/tests/data/pusht/stats.pth b/tests/data/pusht/stats.pth
new file mode 100644
index 00000000..334a01be
Binary files /dev/null and b/tests/data/pusht/stats.pth differ