From a89bea28e1b2f9f6f4353dd65b09c757252029ab Mon Sep 17 00:00:00 2001 From: Nicolas Rabault Date: Mon, 29 Jun 2026 09:12:14 +0200 Subject: [PATCH] refactor(train): hoist _remote_target_in_argv imports to module top Move `import sys` and `from lerobot.configs import JobConfig` out of the function body and into the top-level import block. --- src/lerobot/scripts/lerobot_train.py | 7 ++----- 1 file changed, 2 insertions(+), 5 deletions(-) diff --git a/src/lerobot/scripts/lerobot_train.py b/src/lerobot/scripts/lerobot_train.py index 52c6783fe..f2a152df9 100644 --- a/src/lerobot/scripts/lerobot_train.py +++ b/src/lerobot/scripts/lerobot_train.py @@ -20,6 +20,7 @@ Requires: pip install 'lerobot[training]' (includes dataset + accelerate + wand import dataclasses import logging +import sys import time from contextlib import nullcontext from pprint import pformat @@ -46,7 +47,7 @@ from lerobot.common.train_utils import ( update_last_checkpoint, ) 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.datasets import EpisodeAwareSampler, compute_sampler_state 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: """True when the CLI requests a remote HF Jobs run (--job.target=).""" - import sys - - from lerobot.configs import JobConfig - target = None args = sys.argv[1:] for i, tok in enumerate(args):