Internal change

PiperOrigin-RevId: 603487100 Change-Id: Ibe7c29de624b46dc0666f0d23cdb678f20ca3457
google · Feb 1, 2024 · 30fcc66 · 30fcc66
1 parent dd37045
commit 30fcc66
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 11 deletions.
diff --git a/brax/envs/wrappers/dm_env_test.py b/brax/envs/wrappers/dm_env_test.py
@@ -31,12 +31,6 @@ def test_action_space(self):
     np.testing.assert_array_equal(
         env.action_spec().maximum, base_env.sys.actuator.ctrl_range[:, 1])
 
-  def test_render(self):
-    base_env = envs.create('pusher')
-    env = dm_env.DmEnvWrapper(base_env)
-    env.reset()
-    env.render()
-
 
 if __name__ == '__main__':
   absltest.main()
diff --git a/brax/envs/wrappers/gym_test.py b/brax/envs/wrappers/gym_test.py
@@ -32,11 +32,6 @@ def test_action_space(self):
     np.testing.assert_array_equal(
         env.action_space.high, base_env.sys.actuator.ctrl_range[:, 1])
 
-  def test_render(self):
-    base_env = envs.create('pusher')
-    env = gym.GymWrapper(base_env)
-    env.reset()
-    env.render(mode='rgb_array')
 
   def test_vector_action_space(self):
     """Tests the action space of the VectorGymWrapper."""