[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
9907f2b041
commit
6882dfd012
|
@ -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)
|
||||
|
|
|
@ -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}",
|
||||
|
|
Loading…
Reference in New Issue