admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-20 16:31:55 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use full MP4 sidecar for episode cache benchmark

Browse Source
This commit is contained in:
Pepijn
2026-06-17 20:22:04 +02:00
parent 34d0495d03
commit bdfe8f8ce9
2 changed files with 11 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
scripts/build_mp4_sidecar.py
+2 -5
View File
@@ -41,7 +41,7 @@ def parse_args() -> argparse.Namespace:
return parser.parse_args()
def push_sidecar(local_path: str, data_root: str, episode_count: int) -> list[str]:
def push_sidecar(local_path: str, data_root: str) -> list[str]:
if not data_root.startswith("hf://"):
return []
@@ -49,9 +49,6 @@ def push_sidecar(local_path: str, data_root: str, episode_count: int) -> list[st
fs = fsspec.filesystem("hf")
remote_dir = f"{data_root.rstrip('/')}/meta/mp4-sidecars"
remote_paths = [f"{remote_dir}/{local.name}"]
alias = f"{remote_dir}/molmoact2-{episode_count}.npz"
if alias not in remote_paths:
remote_paths.append(alias)
for remote in remote_paths:
fs.put(str(local), remote)
@@ -87,7 +84,7 @@ def main() -> None:
if args.no_push:
print("push_skipped: --no-push")
else:
pushed = push_sidecar(args.output, args.data_root, total)
pushed = push_sidecar(args.output, args.data_root)
for remote in pushed:
print(f"pushed {remote}")