Remove b/c workaround

This commit is contained in:
Alexander Soare
2024-03-27 12:03:19 +00:00
parent cb6d1e0871
commit 68d02c80cf
2 changed files with 1 additions and 7 deletions

View File

@@ -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,