From c094f40868862cd3403edccc7c934ba17e8060a6 Mon Sep 17 00:00:00 2001 From: pepijn Date: Thu, 11 Jun 2026 14:58:15 +0000 Subject: [PATCH] style: apply ruff-format to lerobot_train.py Collapse the compute_sampler_state(...) call onto one line so the ruff-format pre-commit hook passes (fixes the failing CI check). Co-authored-by: Cursor --- src/lerobot/scripts/lerobot_train.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/src/lerobot/scripts/lerobot_train.py b/src/lerobot/scripts/lerobot_train.py index 469d38b62..bf056b05b 100644 --- a/src/lerobot/scripts/lerobot_train.py +++ b/src/lerobot/scripts/lerobot_train.py @@ -410,9 +410,7 @@ def train(cfg: TrainPipelineConfig, accelerator: "Accelerator | None" = None): f"written with num_processes={saved_num_processes}. The data order resumes at the " "right epoch/offset, but per-rank sample-exactness requires the same world size." ) - sampler_state = compute_sampler_state( - step, len(sampler), cfg.batch_size, ckpt_num_processes - ) + sampler_state = compute_sampler_state(step, len(sampler), cfg.batch_size, ckpt_num_processes) sampler.load_state_dict(sampler_state) if is_main_process: logging.info(