Switch to gymnasium from gym

magent2/environments/adversarial_pursuit/adversarial_pursuit.py

@@ -110,7 +110,7 @@
 """
 
 import numpy as np
-from gym.utils import EzPickle
+from gymnasium.utils import EzPickle
 from pettingzoo.utils.conversions import parallel_to_aec_wrapper
 
 import magent2

magent2/environments/battle/battle.py

@@ -123,7 +123,7 @@
 import math
 
 import numpy as np
-from gym.utils import EzPickle
+from gymnasium.utils import EzPickle
 from pettingzoo.utils.conversions import parallel_to_aec_wrapper
 
 import magent2

magent2/environments/battlefield/battlefield.py

@@ -125,7 +125,7 @@
 import math
 
 import numpy as np
-from gym.utils import EzPickle
+from gymnasium.utils import EzPickle
 from pettingzoo.utils.conversions import parallel_to_aec_wrapper
 
 import magent2

magent2/environments/combined_arms/combined_arms.py

@@ -128,7 +128,7 @@
 import math
 
 import numpy as np
-from gym.utils import EzPickle
+from gymnasium.utils import EzPickle
 from pettingzoo.utils.conversions import parallel_to_aec_wrapper
 
 import magent2

magent2/environments/gather/gather.py

@@ -112,7 +112,7 @@
 """
 
 import numpy as np
-from gym.utils import EzPickle
+from gymnasium.utils import EzPickle
 from pettingzoo.utils.conversions import parallel_to_aec_wrapper
 
 import magent2

magent2/environments/magent_env.py

@@ -1,6 +1,6 @@
 import numpy as np
-from gym.spaces import Box, Discrete
-from gym.utils import seeding
+from gymnasium.spaces import Box, Discrete
+from gymnasium.utils import seeding
 from pettingzoo.utils import wrappers
 from pettingzoo.utils.env import ParallelEnv
 
@@ -140,7 +140,7 @@ def _calc_state_shape(self):
 
     def render(self):
         if self.render_mode is None:
-            # gym.logger.WARN(
+            # gymnasium.logger.WARN(
             #     "You are calling render method without specifying any render mode."
             # )
             return

magent2/environments/tiger_deer/tiger_deer.py

@@ -95,7 +95,7 @@
 """
 
 import numpy as np
-from gym.utils import EzPickle
+from gymnasium.utils import EzPickle
 from pettingzoo.utils.conversions import parallel_to_aec_wrapper
 
 import magent2