diff --git a/brax/envs/wrappers/dm_env_test.py b/brax/envs/wrappers/dm_env_test.py index d65be3d7..7724adc0 100644 --- 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 index aea7f31e..6e668847 100644 --- 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."""