From 30fcc66f9ef4ba42fa4f09c094c814b3a7993b48 Mon Sep 17 00:00:00 2001 From: Brax Team Date: Thu, 1 Feb 2024 22:42:53 +0000 Subject: [PATCH] Internal change PiperOrigin-RevId: 603487100 Change-Id: Ibe7c29de624b46dc0666f0d23cdb678f20ca3457 --- brax/envs/wrappers/dm_env_test.py | 6 ------ brax/envs/wrappers/gym_test.py | 5 ----- 2 files changed, 11 deletions(-) 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."""