[misc] Fix acrobot example eval max horizon.

duburcqa · Nov 26, 2024 · 59a28a7 · 59a28a7
1 parent a7cf740
commit 59a28a7
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/python/gym_jiminy/examples/rllib/acrobot_ppo.py b/python/gym_jiminy/examples/rllib/acrobot_ppo.py
@@ -285,4 +285,7 @@
     rl_module = build_module_from_checkpoint(checkpoint_path)
     policy_fn = build_module_wrapper(rl_module)
     for seed in (1, 1, 2):
-        env.evaluate(policy_fn, seed=seed)  # type: ignore[attr-defined]
+        env.evaluate(  # type: ignore[attr-defined]
+            policy_fn,
+            seed=seed,
+            horizon=env.spec.max_episode_steps)  # type: ignore[union-attr]