From f71b5841182bdca9a4e068f3bf3e31e57235015d Mon Sep 17 00:00:00 2001 From: Scott Lowder Date: Sat, 25 Nov 2023 16:27:24 -0800 Subject: [PATCH] Fix test assertion in wrappers_test.py Replace call to assertTrue with assertEqual --- tf_agents/environments/wrappers_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tf_agents/environments/wrappers_test.py b/tf_agents/environments/wrappers_test.py index 3a07b88d1..1dcfbc9a3 100644 --- a/tf_agents/environments/wrappers_test.py +++ b/tf_agents/environments/wrappers_test.py @@ -289,7 +289,7 @@ def test_fixed_length(self): time_step = env.step(np.array(1, dtype=np.int32)) num_steps += 1 # Verify episode length. - self.assertTrue(num_steps, env.fix_length) + self.assertEqual(num_steps, env.fix_length) # it should automatically reset. self.assertTrue(time_step.is_first())