From 1dbcf584d653e5833c91a2385b83aa872e391613 Mon Sep 17 00:00:00 2001 From: Ville Kuosmanen Date: Sun, 13 Apr 2025 12:08:52 +0200 Subject: [PATCH] lint --- lerobot/scripts/train.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/lerobot/scripts/train.py b/lerobot/scripts/train.py index b73b1171..468e7a1a 100644 --- a/lerobot/scripts/train.py +++ b/lerobot/scripts/train.py @@ -137,10 +137,7 @@ def train(cfg: TrainPipelineConfig): eval_env = make_env(cfg.env, n_envs=cfg.eval.batch_size, use_async_envs=cfg.eval.use_async_envs) logging.info("Creating policy") - if isinstance(dataset, MultiLeRobotDataset): - ds_meta = dataset._datasets[0].meta - else: - ds_meta = dataset.meta + ds_meta = dataset._datasets[0].meta if isinstance(dataset, MultiLeRobotDataset) else dataset.meta policy = make_policy( cfg=cfg.policy, ds_meta=ds_meta,