oops

2024-05-04 11:46:23 +00:00 · 2024-05-04 11:46:23 +00:00 · 9451a10c00
parent 1fb103c2d4
commit 9451a10c00
1 changed files with 1 additions and 1 deletions
--- a/lerobot/scripts/visualize_dataset.py
+++ b/lerobot/scripts/visualize_dataset.py
@ -153,7 +153,7 @@ def visualize_dataset(
            if "next.success" in batch:
                rr.log("next.success", rr.Scalar(batch["next.success"][i].item()))

-    if mode == "local":
+    if mode == "local" and save:
        # save .rrd locally
        output_dir = Path(output_dir)
        output_dir.mkdir(parents=True, exist_ok=True)