diff --git a/lerobot/scripts/visualize_dataset.py b/lerobot/scripts/visualize_dataset.py index 0142881c..ee1e615a 100644 --- 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)