Set seed to 1 for the bsuite catch environment in search_test.py.

This test relies on deterministic environment behavior, so setting the seed will ensure this. PiperOrigin-RevId: 328301292 Change-Id: I432ce7aed2459c328267e1a813a2f309fdddd2bf
google-deepmind · Aug 25, 2020 · 5537cc8 · 5537cc8
1 parent 6b1b413
commit 5537cc8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/acme/agents/tf/mcts/search_test.py b/acme/agents/tf/mcts/search_test.py
@@ -34,7 +34,7 @@ class TestSearch(parameterized.TestCase):
       'bfs',
   ])
   def test_catch(self, policy_type: Text):
-    env = catch.Catch(rows=2)
+    env = catch.Catch(rows=2, seed=1)
     num_actions = env.action_spec().num_values
     model = simulator.Simulator(env)
     eval_fn = lambda _: (np.ones(num_actions) / num_actions, 0.)