remove warning for real worl exps

2024-05-31 10:37:10 +00:00 · 2024-05-31 10:37:10 +00:00 · fa7ebc2d2d
parent 8366648a8a
commit fa7ebc2d2d
1 changed files with 5 additions and 1 deletions
--- a/lerobot/common/datasets/factory.py
+++ b/lerobot/common/datasets/factory.py
@ -55,7 +55,11 @@ def make_dataset(cfg, split: str = "train") -> LeRobotDataset | MultiLeRobotData
            "strings to load multiple datasets."
        )

-    if isinstance(cfg.dataset_repo_id, str) and cfg.env.name not in cfg.dataset_repo_id:
+    if (
+        cfg.env.name != "dora"
+        and isinstance(cfg.dataset_repo_id, str)
+        and cfg.env.name not in cfg.dataset_repo_id
+    ):
        logging.warning(
            f"There might be a mismatch between your training dataset ({cfg.dataset_repo_id=}) and your "
            f"environment ({cfg.env.name=})."