fix tests

2024-03-06 13:55:12 +00:00
parent c2c0ef9927
commit 524d29aa80
4 changed files with 23 additions and 14 deletions
--- a/lerobot/common/envs/factory.py
+++ b/lerobot/common/envs/factory.py
@@ -9,6 +9,7 @@ def make_env(cfg, transform=None):
        "image_size": cfg.env.image_size,
        # TODO(rcadene): do we want a specific eval_env_seed?
        "seed": cfg.seed,
+        "num_prev_obs": cfg.n_obs_steps - 1,
    }

    if cfg.env.name == "simxarm":