From 675476a9efe987c1f50a2af90210218f3f127c96 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Tue, 21 May 2024 17:45:27 +0100 Subject: [PATCH 1/2] Update lerobot/common/datasets/factory.py Co-authored-by: Remi --- lerobot/common/datasets/factory.py | 1 + 1 file changed, 1 insertion(+) diff --git a/lerobot/common/datasets/factory.py b/lerobot/common/datasets/factory.py index 66114557..7bdc2ca9 100644 --- a/lerobot/common/datasets/factory.py +++ b/lerobot/common/datasets/factory.py @@ -31,6 +31,7 @@ def resolve_delta_timestamps(cfg): if delta_timestamps is not None: for key in delta_timestamps: if isinstance(delta_timestamps[key], str): + # TODO(rcadene, alexander-soare): remove `eval` to avoid exploit cfg.training.delta_timestamps[key] = eval(delta_timestamps[key]) From 452d15da6ca9ac2e45620e7b6f784311856f101c Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Tue, 21 May 2024 17:45:41 +0100 Subject: [PATCH 2/2] Update lerobot/scripts/train.py Co-authored-by: Remi --- lerobot/scripts/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/scripts/train.py b/lerobot/scripts/train.py index b7308273..3f1e780c 100644 --- a/lerobot/scripts/train.py +++ b/lerobot/scripts/train.py @@ -231,7 +231,7 @@ def train(cfg: DictConfig, out_dir: str | None = None, job_name: str | None = No if cfg.resume: if not logger.last_checkpoint_dir.exists(): raise RuntimeError( - f"You have set resume=True, but there is no model checpoint in {logger.last_checkpoint_dir}." + f"You have set resume=True, but there is no model checkpoint in {logger.last_checkpoint_dir}." ) else: checkpoint_cfg_path = str(logger.last_pretrained_model_dir / "config.yaml")