From 9451a10c008155344381a910c05b25fab3befee6 Mon Sep 17 00:00:00 2001 From: Cadene Date: Sat, 4 May 2024 11:46:23 +0000 Subject: [PATCH] oops --- lerobot/scripts/visualize_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)