Remove ppo test assertion, only increments during tracing and is error prone.

btaba · copybara-github · commit 9b7af2a8f40e · 2025-11-07T11:24:07.000-08:00
PiperOrigin-RevId: 829512449
Change-Id: If2dbd866253db8c112a62b6ba887cd5aa76a6385
diff --git a/brax/training/agents/ppo/train_test.py b/brax/training/agents/ppo/train_test.py
@@ -54,8 +54,6 @@ def testTrain(self, obs_mode):
         normalize_advantage=False,
     )
     self.assertGreater(metrics['eval/episode_reward'], 135)
-    self.assertEqual(fast.reset_count, 4)  # type: ignore
-    self.assertEqual(fast.step_count, 3)  # type: ignore
 
   @parameterized.parameters(
       ('normal', 'scalar'),

Original file line number	Diff line number	Diff line change
`@@ -54,8 +54,6 @@ def testTrain(self, obs_mode):`
`54`	`54`	`normalize_advantage=False,`
`55`	`55`	`)`
`56`	`56`	`self.assertGreater(metrics['eval/episode_reward'], 135)`
`57`		`- self.assertEqual(fast.reset_count, 4) # type: ignore`
`58`		`- self.assertEqual(fast.step_count, 3) # type: ignore`
`59`	`57`
`60`	`58`	`@parameterized.parameters(`
`61`	`59`	`('normal', 'scalar'),`