From 5537cc896935e647d01d90d13308a9aa627fb6fc Mon Sep 17 00:00:00 2001 From: Gabriel Barth-Maron Date: Tue, 25 Aug 2020 03:30:33 -0700 Subject: [PATCH] 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 --- acme/agents/tf/mcts/search_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/acme/agents/tf/mcts/search_test.py b/acme/agents/tf/mcts/search_test.py index 6b8f1a22c8..f340adc88b 100644 --- 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.)