mirror of
https://github.com/huggingface/lerobot.git
synced 2026-07-20 08:21:50 +00:00
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:
@@ -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",
|
||||||
|
|||||||
@@ -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:]
|
||||||
|
|||||||
@@ -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
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user