From 602fb7bf36a405102a8320f64a8f6109420155d8 Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Thu, 27 Nov 2025 14:09:56 +0100 Subject: [PATCH] remove white lines --- src/lerobot/processor/migrate_policy_normalization.py | 1 + src/lerobot/scripts/lerobot_eval.py | 2 ++ src/lerobot/scripts/lerobot_train.py | 2 +- 3 files changed, 4 insertions(+), 1 deletion(-) diff --git a/src/lerobot/processor/migrate_policy_normalization.py b/src/lerobot/processor/migrate_policy_normalization.py index dd7ecba4a..525b7431c 100644 --- a/src/lerobot/processor/migrate_policy_normalization.py +++ b/src/lerobot/processor/migrate_policy_normalization.py @@ -331,6 +331,7 @@ def load_state_dict_with_missing_key_handling( # Check for missing keys missing_keys = load_result.missing_keys unexpected_keys = load_result.unexpected_keys + # Filter out whitelisted missing keys policy_type_lower = policy_type.lower() whitelisted_keys = known_missing_keys_whitelist.get(policy_type_lower, []) diff --git a/src/lerobot/scripts/lerobot_eval.py b/src/lerobot/scripts/lerobot_eval.py index db93dbe0d..b6982a927 100644 --- a/src/lerobot/scripts/lerobot_eval.py +++ b/src/lerobot/scripts/lerobot_eval.py @@ -175,6 +175,7 @@ def rollout( with torch.inference_mode(): action = policy.select_action(observation) action = postprocessor(action) + action_transition = {"action": action} action_transition = env_postprocessor(action_transition) action = action_transition["action"] @@ -510,6 +511,7 @@ def eval_main(cfg: EvalPipelineConfig): envs = make_env(cfg.env, n_envs=cfg.eval.batch_size, use_async_envs=cfg.eval.use_async_envs) logging.info("Making policy.") + policy = make_policy( cfg=cfg.policy, env_cfg=cfg.env, diff --git a/src/lerobot/scripts/lerobot_train.py b/src/lerobot/scripts/lerobot_train.py index 92fd1917d..96ec0f781 100644 --- a/src/lerobot/scripts/lerobot_train.py +++ b/src/lerobot/scripts/lerobot_train.py @@ -200,12 +200,12 @@ def train(cfg: TrainPipelineConfig, accelerator: Accelerator | None = None): if is_main_process: logging.info("Creating policy") - policy = make_policy( cfg=cfg.policy, ds_meta=dataset.meta, rename_map=cfg.rename_map, ) + # Wait for all processes to finish policy creation before continuing accelerator.wait_for_everyone()