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 dec1c6ddc2
commit 62badcba4e
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 (
@@ -55,6 +55,7 @@ __all__ = [
"DatasetRecordConfig", "DatasetRecordConfig",
"DatasetConfig", "DatasetConfig",
"EvalConfig", "EvalConfig",
"JobConfig",
"MessageTurn", "MessageTurn",
"PeftConfig", "PeftConfig",
"PreTrainedConfig", "PreTrainedConfig",
+1 -1
View File
@@ -751,7 +751,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