Add pusht test artifact

This commit is contained in:
Simon Alibert
2024-03-08 15:54:30 +01:00
parent 7dbdbb051c
commit 89eaab140b
19 changed files with 42 additions and 0 deletions

1
tests/data/meta.json Normal file
View File

@@ -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'>"}