Make random_mdp test deterministic by seeding the environment.

HumanCompatibleAI · Sep 5, 2023 · 4872ceb · 4872ceb
1 parent 09c5f2f
commit 4872ceb
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/algorithms/test_mce_irl.py b/tests/algorithms/test_mce_irl.py
@@ -132,6 +132,8 @@ def test_infinite_horizon_error(random_mdp, rng):
 def test_policy_om_random_mdp(discount: float):
     """Test that optimal policy occupancy measure ("om") for a random MDP is sane."""
     mdp = gym.make("seals/Random-v0")
+    mdp.seed(0)
+
     V, Q, pi = mce_partition_fh(mdp, discount=discount)
     assert np.all(np.isfinite(V))
     assert np.all(np.isfinite(Q))