From aa03a27f0aea35ff2783b5a8940998aa72f42af0 Mon Sep 17 00:00:00 2001 From: Remi Cadene Date: Wed, 27 Nov 2024 16:45:07 +0100 Subject: [PATCH] fix --- lerobot/scripts/visualize_dataset_html.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/lerobot/scripts/visualize_dataset_html.py b/lerobot/scripts/visualize_dataset_html.py index 39ce81e6e..1799e4df9 100644 --- a/lerobot/scripts/visualize_dataset_html.py +++ b/lerobot/scripts/visualize_dataset_html.py @@ -88,7 +88,7 @@ def run_server( port: str, static_folder: Path, template_folder: Path, - has_policy = False, + has_policy=False, ): app = Flask(__name__, static_folder=static_folder.resolve(), template_folder=template_folder.resolve()) app.config["SEND_FILE_MAX_AGE_DEFAULT"] = 0 # specifying not to cache @@ -131,7 +131,7 @@ def show_episode(dataset_namespace, dataset_name, episode_id): dataset_info=dataset_info, videos_info=videos_info, ep_csv_url=ep_csv_url, - has_policy = has_policy, + has_policy=has_policy, ) app.run(host=host, port=port) @@ -404,7 +404,7 @@ def main(): args = parser.parse_args() kwargs = vars(args) repo_id = kwargs.pop("repo_id") - # root = kwargs.pop("root") + kwargs.pop("root") dataset = LeRobotDataset(repo_id) visualize_dataset_html(dataset, **kwargs)