hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

remove __name__ outside script

Browse Source
This commit is contained in:
Cadene
2024-04-10 17:16:44 +00:00
parent f8c5a2eb10
commit 3914831585
1 changed files with 0 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lerobot/common/datasets/pusht.py
View File

@@ -209,16 +209,3 @@ class PushtDataset(torch.utils.data.Dataset):
self.data_dict[key] = torch.cat([x[key] for x in ep_dicts])
self.data_dict["index"] = torch.arange(0, total_frames, 1)
if __name__ == "__main__":
dataset = PushtDataset(
"pusht",
root=Path("data"),
delta_timestamps={
"observation.image": [0, -1, -0.2, -0.1],
"observation.state": [0, -1, -0.2, -0.1],
"action": [-0.1, 0, 1, 2, 3],
},
)
dataset[10]