From a6c16016b44650c2bd1d403e096ce682ec197dab Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Antonio=20Serrano=20Mu=C3=B1oz?= Date: Tue, 13 Aug 2024 13:20:25 -0400 Subject: [PATCH] Remove print statements --- tests/jax/test_jax_wrapper_gym.py | 2 -- tests/jax/test_jax_wrapper_gymnasium.py | 2 -- 2 files changed, 4 deletions(-) diff --git a/tests/jax/test_jax_wrapper_gym.py b/tests/jax/test_jax_wrapper_gym.py index 11f00b95..29a95177 100644 --- a/tests/jax/test_jax_wrapper_gym.py +++ b/tests/jax/test_jax_wrapper_gym.py @@ -39,8 +39,6 @@ def test_env(capsys: pytest.CaptureFixture, backend: str): # check methods for _ in range(2): observation, info = env.reset() - with capsys.disabled(): - print(observation.shape, type(observation.shape)) assert isinstance(observation, Array) and observation.shape == (num_envs, 3) assert isinstance(info, Mapping) for _ in range(3): diff --git a/tests/jax/test_jax_wrapper_gymnasium.py b/tests/jax/test_jax_wrapper_gymnasium.py index e221db99..3375d8bc 100644 --- a/tests/jax/test_jax_wrapper_gymnasium.py +++ b/tests/jax/test_jax_wrapper_gymnasium.py @@ -39,8 +39,6 @@ def test_env(capsys: pytest.CaptureFixture, backend: str): # check methods for _ in range(2): observation, info = env.reset() - with capsys.disabled(): - print(observation.shape, type(observation.shape)) assert isinstance(observation, Array) and observation.shape == (num_envs, 3) assert isinstance(info, Mapping) for _ in range(3):