From 16555ff74d78ff4935a7bbce154094dedbb0fc12 Mon Sep 17 00:00:00 2001
From: Kallinteris Andreas
 <30759571+Kallinteris-Andreas@users.noreply.github.com>
Date: Fri, 11 Oct 2024 18:10:04 +0300
Subject: [PATCH 1/2] Update test_envs.py

---
 tests/test_envs.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/tests/test_envs.py b/tests/test_envs.py
index b018f896..04c119d1 100644
--- a/tests/test_envs.py
+++ b/tests/test_envs.py
@@ -8,8 +8,11 @@
 from gymnasium.error import Error
 from gymnasium.utils.env_checker import check_env, data_equivalence
 
+import gymnasium_robotics
 from tests.utils import all_testing_env_specs, assert_equals
 
+gym.register_envs(gymnasium_robotics)
+
 CHECK_ENV_IGNORE_WARNINGS = [
     f"\x1b[33mWARN: {message}\x1b[0m"
     for message in [

From bed003e5e93570305d599cbd03927482268c632e Mon Sep 17 00:00:00 2001
From: Kallinteris Andreas
 <30759571+Kallinteris-Andreas@users.noreply.github.com>
Date: Fri, 11 Oct 2024 18:20:59 +0300
Subject: [PATCH 2/2] Update test_envs.py

---
 tests/test_envs.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/test_envs.py b/tests/test_envs.py
index 04c119d1..f880b297 100644
--- a/tests/test_envs.py
+++ b/tests/test_envs.py
@@ -30,7 +30,7 @@
 
 
 @pytest.mark.parametrize(
-    "spec", all_testing_env_specs, ids=[spec.id for spec in all_testing_env_specs]
+    "spec", non_mujoco_py_env_specs, ids=[spec.id for spec in non_mujoco_py_env_specs]
 )
 def test_env(spec):
     # Capture warnings
@@ -53,7 +53,7 @@ def test_env(spec):
 
 
 @pytest.mark.parametrize(
-    "env_spec", all_testing_env_specs, ids=[env.id for env in all_testing_env_specs]
+    "env_spec", non_mujoco_py_env_specs, ids=[env.id for env in non_mujoco_py_env_specs]
 )
 def test_env_determinism_rollout(env_spec: EnvSpec):
     """Run a rollout with two environments and assert equality.