From ab687f5ca8973405ac8464848639ba2ed73e45d2 Mon Sep 17 00:00:00 2001 From: yangchen73 <122090643@link.cuhk.edu.cn> Date: Thu, 26 Feb 2026 05:25:03 +0000 Subject: [PATCH 1/2] Fix: allow num_envs override from trainer config --- embodichain/agents/rl/train.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/embodichain/agents/rl/train.py b/embodichain/agents/rl/train.py index 32bc0383..4b88c3d3 100644 --- a/embodichain/agents/rl/train.py +++ b/embodichain/agents/rl/train.py @@ -137,6 +137,9 @@ def train_from_config(config_path: str): gym_config_data, manager_modules=DEFAULT_MANAGER_MODULES ) + if num_envs is not None: + gym_env_cfg.num_envs = num_envs + # Ensure sim configuration mirrors runtime overrides if gym_env_cfg.sim_cfg is None: gym_env_cfg.sim_cfg = SimulationManagerCfg() From abca31e12006ba3d6046068abdc8e653d3ed1d3f Mon Sep 17 00:00:00 2001 From: yangchen73 <122090643@link.cuhk.edu.cn> Date: Thu, 26 Feb 2026 05:26:28 +0000 Subject: [PATCH 2/2] reformat file --- embodichain/agents/rl/train.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/embodichain/agents/rl/train.py b/embodichain/agents/rl/train.py index 4b88c3d3..1c520000 100644 --- a/embodichain/agents/rl/train.py +++ b/embodichain/agents/rl/train.py @@ -139,7 +139,7 @@ def train_from_config(config_path: str): if num_envs is not None: gym_env_cfg.num_envs = num_envs - + # Ensure sim configuration mirrors runtime overrides if gym_env_cfg.sim_cfg is None: gym_env_cfg.sim_cfg = SimulationManagerCfg()