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 .dataset import DatasetRecordConfig
from .default import DatasetConfig, EvalConfig, PeftConfig, WandBConfig from .default import DatasetConfig, EvalConfig, JobConfig, PeftConfig, WandBConfig
from .policies import PreTrainedConfig from .policies import PreTrainedConfig
from .recipe import MessageTurn, TrainingRecipe, load_recipe from .recipe import MessageTurn, TrainingRecipe, load_recipe
from .types import ( from .types import (
@@ -50,6 +50,7 @@ __all__ = [
"DatasetRecordConfig", "DatasetRecordConfig",
"DatasetConfig", "DatasetConfig",
"EvalConfig", "EvalConfig",
"JobConfig",
"MessageTurn", "MessageTurn",
"PeftConfig", "PeftConfig",
"PreTrainedConfig", "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>).""" """True when the CLI requests a remote HF Jobs run (--job.target=<non-local>)."""
import sys import sys
from lerobot.configs.default import JobConfig from lerobot.configs import JobConfig
target = None target = None
args = sys.argv[1:] args = sys.argv[1:]
+1 -1
View File
@@ -15,7 +15,7 @@
import draccus import draccus
import pytest import pytest
from lerobot.configs.default import JobConfig from lerobot.configs import JobConfig
from lerobot.configs.train import TrainPipelineConfig from lerobot.configs.train import TrainPipelineConfig