Sheep training flock _ improver

2026-04-25 17:07:03 +01:00
parent 3a5decb185
commit cc6d72e472
2 changed files with 47 additions and 18 deletions
@@ -114,9 +114,10 @@ def reward_cfg(cfg: dict) -> dict:
    return {k: v for k, v in cfg.items() if k != "ent_coef"}


-def make_env(n_sheep, seed, max_steps, rcfg):
+def make_env(n_sheep, seed, max_steps, rcfg, random_n_sheep=False):
    def _init():
-        env = HerdingEnv(n_sheep=n_sheep, max_steps=max_steps, reward_cfg=rcfg)
+        env = HerdingEnv(n_sheep=n_sheep, max_steps=max_steps,
+                         reward_cfg=rcfg, random_n_sheep=random_n_sheep)
        env.reset(seed=seed)
        return env
    return _init