From 68d02c80cff25cb8852a4c30e3d2162d7f2b2de5 Mon Sep 17 00:00:00 2001 From: Alexander Soare Date: Wed, 27 Mar 2024 12:03:19 +0000 Subject: [PATCH] Remove b/c workaround --- README.md | 1 - lerobot/common/datasets/factory.py | 7 +------ 2 files changed, 1 insertion(+), 7 deletions(-) diff --git a/README.md b/README.md index 35911869..c621897a 100644 --- a/README.md +++ b/README.md @@ -145,7 +145,6 @@ Or you can achieve the same result by executing our script from the command line ```bash python lerobot/scripts/eval.py \ --hub-id lerobot/diffusion_policy_pusht_image \ ---revision v1.0 \ eval_episodes=10 \ hydra.run.dir=outputs/eval/example_hub ``` diff --git a/lerobot/common/datasets/factory.py b/lerobot/common/datasets/factory.py index 4212e023..04077034 100644 --- a/lerobot/common/datasets/factory.py +++ b/lerobot/common/datasets/factory.py @@ -81,13 +81,8 @@ def make_offline_buffer( else: raise ValueError(cfg.env.name) - # TODO(rcadene): backward compatiblity to load pretrained pusht policy - dataset_id = cfg.get("dataset_id") - if dataset_id is None and cfg.env.name == "pusht": - dataset_id = "pusht" - offline_buffer = clsfunc( - dataset_id=dataset_id, + dataset_id=cfg.dataset_id, sampler=sampler, batch_size=batch_size, root=DATA_DIR,