Remove dependence of wrapper on old gym

2023-07-07 13:11:52 +02:00 · 2023-07-07 13:11:52 +02:00 · 27f8335a0d
commit 27f8335a0d
parent 14a95eb5ca
1 changed files with 4 additions and 1 deletions
--- a/fancy_gym/utils/wrappers.py
+++ b/fancy_gym/utils/wrappers.py
@ -1,5 +1,8 @@
 from gymnasium.spaces import Box, Dict, flatten, flatten_space
-from gym.spaces import Box as OldBox
+try:
+    from gym.spaces import Box as OldBox
+except ImportError:
+    OldBox = None
 import gymnasium as gym
 import numpy as np
 import copy