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

oops

This commit is contained in:
Cadene 2024-05-04 11:46:23 +00:00
parent 1fb103c2d4
commit 9451a10c00
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lerobot/scripts/visualize_dataset.py
View File

@ -153,7 +153,7 @@ def visualize_dataset(
if "next.success" in batch: if "next.success" in batch:
rr.log("next.success", rr.Scalar(batch["next.success"][i].item())) rr.log("next.success", rr.Scalar(batch["next.success"][i].item()))
if mode == "local": if mode == "local" and save:
# save .rrd locally # save .rrd locally
output_dir = Path(output_dir) output_dir = Path(output_dir)
output_dir.mkdir(parents=True, exist_ok=True) output_dir.mkdir(parents=True, exist_ok=True)