Prevent resuming from hub (#726)

huggingface · Feb 13, 2025 · 1e49cc4 · 1e49cc4
1 parent e710959
commit 1e49cc4
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/lerobot/configs/train.py b/lerobot/configs/train.py
@@ -85,6 +85,11 @@ def validate(self):
             config_path = parser.parse_arg("config_path")
             if not config_path:
                 raise ValueError("A config_path is expected when resuming a run.")
+            if not Path(config_path).resolve().exists():
+                raise NotADirectoryError(
+                    f"{config_path=} is expected to be a local path. "
+                    "Resuming from the hub is not supported for now."
+                )
             policy_path = Path(config_path).parent
             self.policy.pretrained_path = policy_path
             self.checkpoint_path = policy_path.parent