refactor(configs): export JobConfig from the configs package

Re-export JobConfig in lerobot/configs/__init__.py so external callers
import it as `from lerobot.configs import JobConfig`, matching the other
config classes. Adapt the train script and test imports.
This commit is contained in:
Nicolas Rabault
2026-06-24 11:04:16 +02:00
parent 6ad1e6b6ae
commit 30cc3d59f5
3 changed files with 4 additions and 3 deletions
+2 -1
View File
@@ -22,7 +22,7 @@ Import them directly: ``from lerobot.configs.train import TrainPipelineConfig``
"""
from .dataset import DatasetRecordConfig
from .default import DatasetConfig, EvalConfig, PeftConfig, WandBConfig
from .default import DatasetConfig, EvalConfig, JobConfig, PeftConfig, WandBConfig
from .policies import PreTrainedConfig
from .recipe import MessageTurn, TrainingRecipe, load_recipe
from .types import (
@@ -50,6 +50,7 @@ __all__ = [
"DatasetRecordConfig",
"DatasetConfig",
"EvalConfig",
"JobConfig",
"MessageTurn",
"PeftConfig",
"PreTrainedConfig",
+1 -1
View File
@@ -693,7 +693,7 @@ def _remote_target_in_argv() -> bool:
"""True when the CLI requests a remote HF Jobs run (--job.target=<non-local>)."""
import sys
from lerobot.configs.default import JobConfig
from lerobot.configs import JobConfig
target = None
args = sys.argv[1:]
+1 -1
View File
@@ -15,7 +15,7 @@
import draccus
import pytest
from lerobot.configs.default import JobConfig
from lerobot.configs import JobConfig
from lerobot.configs.train import TrainPipelineConfig