For python 3.12, importorskip bsuite

Farama-Foundation · May 21, 2024 · c5607be · c5607be
1 parent d824111
commit c5607be
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/setup.py b/setup.py
@@ -49,7 +49,7 @@ def get_version():
         "pettingzoo>=1.23",
         "dm-meltingpot>=2.2.2; python_version > '3.10'",
     ],
-    "bsuite": ["bsuite>=0.3.5"],
+    "bsuite": ["bsuite>=0.3.5; python_version < '3.12'"],
 }
 extras["all"] = [
     lib for key, libs in extras.items() if key != "gym-v21" for lib in libs

diff --git a/tests/test_bsuite.py b/tests/test_bsuite.py
@@ -2,7 +2,6 @@
 import pickle
 import warnings
 
-import bsuite
 import gymnasium as gym
 import pytest
 from gymnasium.envs.registration import registry
@@ -13,6 +12,8 @@
 
 gym.register_envs(shimmy)
 
+bsuite = pytest.importorskip("bsuite")
+
 BSUITE_ENV_IDS = [
     env_id
     for env_id in registry