Hotfix test_examples.py (#87)
This commit is contained in:
parent
8d980940a2
commit
ac0ab27333
|
@ -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=}")
|
||||
|
|
|
@ -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,
|
||||
)
|
||||
|
|
Loading…
Reference in New Issue