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

Remove image_writer arg

This commit is contained in:
Simon Alibert 2024-10-29 16:07:10 +01:00
parent 8d57093ce7
commit fee5fa5c2e
1 changed files with 2 additions and 1 deletions
lerobot/common/datasets
lerobot_dataset.py

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

@ -78,7 +78,6 @@ class LeRobotDataset(torch.utils.data.Dataset):
download_videos: bool = True, download_videos: bool = True,
local_files_only: bool = False, local_files_only: bool = False,
video_backend: str | None = None, video_backend: str | None = None,
image_writer: ImageWriter | None = None,
): ):
"""LeRobotDataset encapsulates 3 main things: """LeRobotDataset encapsulates 3 main things:
- metadata: - metadata:
@ -168,6 +167,8 @@ class LeRobotDataset(torch.utils.data.Dataset):
download_videos (bool, optional): Flag to download the videos. Note that when set to True but the download_videos (bool, optional): Flag to download the videos. Note that when set to True but the
video files are already present on local disk, they won't be downloaded again. Defaults to video files are already present on local disk, they won't be downloaded again. Defaults to
True. True.
local_files_only (bool, optional): Flag to use local files only. If True, no requests to the hub
will be made. Defaults to False.
video_backend (str | None, optional): Video backend to use for decoding videos. There is currently video_backend (str | None, optional): Video backend to use for decoding videos. There is currently
a single option which is the pyav decoder used by Torchvision. Defaults to pyav. a single option which is the pyav decoder used by Torchvision. Defaults to pyav.
""" """