admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-24 02:06:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

refactor(train): hoist _remote_target_in_argv imports to module top

Browse Source 
Move `import sys` and `from lerobot.configs import JobConfig` out of the
function body and into the top-level import block.
This commit is contained in:
Nicolas Rabault
2026-06-29 09:12:14 +02:00
parent f2b1e62dda
commit a89bea28e1
1 changed files with 2 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/lerobot_train.py
+2 -5
View File
@@ -20,6 +20,7 @@ Requires: pip install 'lerobot[training]' (includes dataset + accelerate + wand
import dataclasses import dataclasses
import logging import logging
import sys
import time import time
from contextlib import nullcontext from contextlib import nullcontext
from pprint import pformat from pprint import pformat
@@ -46,7 +47,7 @@ from lerobot.common.train_utils import (
update_last_checkpoint, update_last_checkpoint,
) )
from lerobot.common.wandb_utils import WandBLogger from lerobot.common.wandb_utils import WandBLogger
from lerobot.configs import parser from lerobot.configs import JobConfig, parser
from lerobot.configs.train import TrainPipelineConfig from lerobot.configs.train import TrainPipelineConfig
from lerobot.datasets import EpisodeAwareSampler, compute_sampler_state from lerobot.datasets import EpisodeAwareSampler, compute_sampler_state
from lerobot.datasets.factory import make_train_eval_datasets from lerobot.datasets.factory import make_train_eval_datasets
@@ -748,10 +749,6 @@ def train(cfg: TrainPipelineConfig, accelerator: "Accelerator | None" = None):
def _remote_target_in_argv() -> bool: 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
from lerobot.configs import JobConfig
target = None target = None
args = sys.argv[1:] args = sys.argv[1:]
for i, tok in enumerate(args): for i, tok in enumerate(args):