From 6882dfd0129a91140f70d1f7b978ab1637ce3db0 Mon Sep 17 00:00:00 2001 From: "pre-commit-ci[bot]" <66853113+pre-commit-ci[bot]@users.noreply.github.com> Date: Sun, 13 Apr 2025 10:07:08 +0000 Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks for more information, see https://pre-commit.ci --- lerobot/common/datasets/factory.py | 6 +++--- lerobot/common/logger.py | 4 ++-- 2 files changed, 5 insertions(+), 5 deletions(-) diff --git a/lerobot/common/datasets/factory.py b/lerobot/common/datasets/factory.py index f0e6abe7..416074fe 100644 --- a/lerobot/common/datasets/factory.py +++ b/lerobot/common/datasets/factory.py @@ -81,8 +81,8 @@ def make_dataset(cfg: TrainPipelineConfig) -> LeRobotDataset | MultiLeRobotDatas image_transforms = ( ImageTransforms(cfg.dataset.image_transforms) if cfg.dataset.image_transforms.enable else None ) - if cfg.dataset.repo_id.startswith('['): - datasets = cfg.dataset.repo_id.strip('[]').split(',') + if cfg.dataset.repo_id.startswith("["): + datasets = cfg.dataset.repo_id.strip("[]").split(",") datasets = [x.strip() for x in datasets] delta_timestamps = {} for ds in datasets: @@ -98,7 +98,7 @@ def make_dataset(cfg: TrainPipelineConfig) -> LeRobotDataset | MultiLeRobotDatas ) logging.info( "Multiple datasets were provided. Applied the following index mapping to the provided datasets: " - f"{pformat(dataset.repo_id_to_index , indent=2)}" + f"{pformat(dataset.repo_id_to_index, indent=2)}" ) else: ds_meta = LeRobotDatasetMetadata(cfg.dataset.repo_id) diff --git a/lerobot/common/logger.py b/lerobot/common/logger.py index 18458ae6..b874f4ba 100644 --- a/lerobot/common/logger.py +++ b/lerobot/common/logger.py @@ -49,8 +49,8 @@ def cfg_to_group(cfg: TrainPipelineConfig, return_list: bool = False) -> list[st """Return a group name for logging. Optionally returns group name as list.""" # TODO: these were used to support multirepodataset in the past - think how they could be supported in new way? dataset_tag = cfg.dataset.repo_id - if dataset_tag.startswith('['): - tags = dataset_tag.strip('[]').split(',') + if dataset_tag.startswith("["): + tags = dataset_tag.strip("[]").split(",") dataset_tag = f"{tags[0].strip()}_and_more" lst = [ f"policy:{cfg.policy.type}",