Update test_envs.py

Farama-Foundation · Oct 11, 2024 · bed003e · bed003e
1 parent 16555ff
commit bed003e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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.