Remove print statements

Toni-SM · Aug 13, 2024 · a6c1601 · a6c1601
1 parent 25e969a
commit a6c1601
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git 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
@@ -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):