fix unit test

2024-09-26 11:48:22 +02:00
parent 500d505bf6
commit f2b1842d69
2 changed files with 4 additions and 3 deletions
--- a/tests/test_control_robot.py
+++ b/tests/test_control_robot.py
@@ -114,6 +114,7 @@ def test_record_and_replay_and_policy(tmpdir, request, robot_type, mock):
        cfg,
        warmup_time_s=1,
        episode_time_s=1,
+        num_episodes=2,
        run_compute_stats=False,
        push_to_hub=False,
        video=False,