update sac test

google · Dec 4, 2024 · 19f6337 · 19f6337
1 parent 692d626
commit 19f6337
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/brax/training/agents/sac/train_test.py b/brax/training/agents/sac/train_test.py
@@ -46,7 +46,7 @@ def testTrain(self):
         num_evals=3,
         seed=0)
     self.assertGreater(metrics['eval/episode_reward'], 140 * 0.995)
-    self.assertEqual(fast.reset_count, 2)  # type: ignore
+    self.assertEqual(fast.reset_count, 3)  # type: ignore
     # once for prefill, once for train, once for eval
     self.assertEqual(fast.step_count, 3)  # type: ignore