wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Revert lerobot datasets dependency

This commit is contained in:
Simon Alibert 2024-04-18 11:03:32 +02:00
parent 73083269cb
commit 673dd8fd3f
3 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
lerobot/common/datasets/aloha.py
View File

@ -3,7 +3,6 @@ from pathlib import Path
import torch import torch
from datasets import load_dataset, load_from_disk from datasets import load_dataset, load_from_disk
import lerobot
from lerobot.common.datasets.utils import load_previous_and_future_frames from lerobot.common.datasets.utils import load_previous_and_future_frames
@ -15,7 +14,12 @@ class AlohaDataset(torch.utils.data.Dataset):
https://huggingface.co/datasets/lerobot/aloha_sim_transfer_cube_scripted https://huggingface.co/datasets/lerobot/aloha_sim_transfer_cube_scripted
""" """
available_datasets = lerobot.available_datasets["aloha"] available_datasets = [
"aloha_sim_insertion_human",
"aloha_sim_insertion_scripted",
"aloha_sim_transfer_cube_human",
"aloha_sim_transfer_cube_scripted",
]
fps = 50 fps = 50
image_keys = ["observation.images.top"] image_keys = ["observation.images.top"]

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

@ -3,7 +3,6 @@ from pathlib import Path
import torch import torch
from datasets import load_dataset, load_from_disk from datasets import load_dataset, load_from_disk
import lerobot
from lerobot.common.datasets.utils import load_previous_and_future_frames from lerobot.common.datasets.utils import load_previous_and_future_frames
@ -18,7 +17,7 @@ class PushtDataset(torch.utils.data.Dataset):
If `None`, no shift is applied to current timestamp and the data from the current frame is loaded. If `None`, no shift is applied to current timestamp and the data from the current frame is loaded.
""" """
available_datasets = lerobot.available_datasets["pusht"] available_datasets = ["pusht"]
fps = 10 fps = 10
image_keys = ["observation.image"] image_keys = ["observation.image"]

3
lerobot/common/datasets/xarm.py
View File

@ -3,7 +3,6 @@ from pathlib import Path
import torch import torch
from datasets import load_dataset, load_from_disk from datasets import load_dataset, load_from_disk
import lerobot
from lerobot.common.datasets.utils import load_previous_and_future_frames from lerobot.common.datasets.utils import load_previous_and_future_frames
@ -12,7 +11,7 @@ class XarmDataset(torch.utils.data.Dataset):
https://huggingface.co/datasets/lerobot/xarm_lift_medium https://huggingface.co/datasets/lerobot/xarm_lift_medium
""" """
available_datasets = lerobot.available_datasets["xarm"] available_datasets = ["xarm_lift_medium"]
fps = 15 fps = 15
image_keys = ["observation.image"] image_keys = ["observation.image"]