Make SQIL tests more deterministic by adding more seeding.

HumanCompatibleAI · Oct 9, 2023 · 5552245 · 5552245
1 parent 959f897
commit 5552245
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/tests/algorithms/test_sqil.py b/tests/algorithms/test_sqil.py
@@ -157,7 +157,7 @@ def test_sqil_few_demonstrations_discrete(
         cartpole_venv,
         "seals/CartPole-v0",
         rl_algo_class=dqn.DQN,
-        rl_kwargs=dict(learning_starts=10),
+        rl_kwargs=dict(learning_starts=10, seed=42),
     )
 
 
@@ -174,6 +174,7 @@ def test_sqil_few_demonstrations_continuous(
         pendulum_single_venv,
         "Pendulum-v1",
         rl_algo_class=rl_algo_class,
+        rl_kwargs=dict(seed=42),
     )