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

fix(examples): remove lint warnings/errors

This commit is contained in:
Steven Palma 2025-03-07 14:26:33 +01:00
parent 254bc707e7
commit 0ed08c0b1f
No known key found for this signature in database
GPG Key ID: F4D348E38D8F8ADD
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
examples/port_datasets/pusht_zarr.py
View File

@ -222,7 +222,7 @@ def main(raw_dir: Path, repo_id: str, mode: str = "video", push_to_hub: bool = T
if __name__ == "__main__": if __name__ == "__main__":
# To try this script, modify the repo id with your own HuggingFace user (e.g cadene/pusht) # To try this script, modify the repo id with your own HuggingFace user (e.g cadene/pusht)
repo_id = "lerobot/pusht" repository_id = "lerobot/pusht"
modes = ["video", "image", "keypoints"] modes = ["video", "image", "keypoints"]
# Uncomment if you want to try with a specific mode # Uncomment if you want to try with a specific mode
@ -230,13 +230,13 @@ if __name__ == "__main__":
# modes = ["image"] # modes = ["image"]
# modes = ["keypoints"] # modes = ["keypoints"]
raw_dir = Path("data/lerobot-raw/pusht_raw") data_dir = Path("data/lerobot-raw/pusht_raw")
for mode in modes: for available_mode in modes:
if mode in ["image", "keypoints"]: if available_mode in ["image", "keypoints"]:
repo_id += f"_{mode}" repository_id += f"_{available_mode}"
# download and load raw dataset, create LeRobotDataset, populate it, push to hub # download and load raw dataset, create LeRobotDataset, populate it, push to hub
main(raw_dir, repo_id=repo_id, mode=mode) main(data_dir, repo_id=repository_id, mode=available_mode)
# Uncomment if you want to load the local dataset and explore it # Uncomment if you want to load the local dataset and explore it
# dataset = LeRobotDataset(repo_id=repo_id) # dataset = LeRobotDataset(repo_id=repo_id)