hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Progress on aggregate_datasets

Browse Source
This commit is contained in:
Remi Cadene
2025-04-19 19:11:53 +05:30
parent 54b5c805bf
commit b0cca75e5e
7 changed files with 149 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tests/fixtures/hub.py vendored
View File

@@ -62,6 +62,7 @@ def mock_snapshot_download_factory(
if episodes is None:
episodes = episodes_factory(
features=info["features"],
fps=info["fps"],
total_episodes=info["total_episodes"],
total_frames=info["total_frames"],
tasks=tasks,
@@ -121,6 +122,7 @@ def mock_snapshot_download_factory(
create_tasks(local_dir, tasks)
if has_episodes:
create_episodes(local_dir, episodes)
# TODO(rcadene): create_videos?
if has_data:
create_hf_dataset(local_dir, hf_dataset)