From 8e5b4365ac02fafe86d517f5bad6d11e38967cc6 Mon Sep 17 00:00:00 2001 From: Cadene Date: Thu, 11 Apr 2024 13:57:22 +0000 Subject: [PATCH] fix --- 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 ed95b39a..4b7b7d6c 100644 --- a/lerobot/scripts/visualize_dataset.py +++ b/lerobot/scripts/visualize_dataset.py @@ -79,7 +79,7 @@ def render_dataset(dataset, out_dir, max_num_episodes): out_dir.mkdir(parents=True, exist_ok=True) for im_key in dataset.image_keys: - if len(dataset.image_keys) > 0: + if len(dataset.image_keys) > 1: im_name = im_key.replace("observation.images.", "") video_path = out_dir / f"episode_{ep_id}_{im_name}.mp4" else: