Hotfix test_examples.py (#87)

This commit is contained in:
Simon Alibert 2024-04-19 12:36:04 +02:00 committed by GitHub
parent 8d980940a2
commit ac0ab27333
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 2 additions and 2 deletions

View File

@ -37,7 +37,7 @@ from datasets import load_dataset
# TODO(rcadene): list available datasets on lerobot page using `datasets`
# download/load hugging face dataset in pyarrow format
hf_dataset, fps = load_dataset("lerobot/pusht", split="train"), 10
hf_dataset, fps = load_dataset("lerobot/pusht", revision="v1.0", split="train"), 10
# display name of dataset and its features
print(f"{hf_dataset=}")

View File

@ -87,7 +87,7 @@ print(f"{dataset[0]['action'].shape=}") # (64,c)
# because they are just PyTorch datasets.
dataloader = torch.utils.data.DataLoader(
dataset,
num_workers=4,
num_workers=0,
batch_size=32,
shuffle=True,
)