Sanitize cfg.env

2024-02-25 12:02:29 +00:00
parent 9b469c4232
commit ed80db2846
6 changed files with 46 additions and 42 deletions
--- a/lerobot/scripts/eval.py
+++ b/lerobot/scripts/eval.py
@@ -126,8 +126,8 @@ def eval(cfg: dict, out_dir=None):
        policy=policy,
        save_video=True,
        video_dir=Path(out_dir) / "eval",
-        fps=cfg.fps,
-        max_steps=cfg.episode_length,
+        fps=cfg.env.fps,
+        max_steps=cfg.env.episode_length,
        num_episodes=cfg.eval_episodes,
    )
    print(metrics)