Merge pull request #4 from dsctt/d/magent2

Switch to gymnasium from gym
Farama-Foundation · Oct 17, 2022 · 736dda5 · 736dda5
2 parents 9a70ee2 + dcc5020
commit 736dda5
Show file tree

Hide file tree

Showing 7 changed files with 9 additions and 9 deletions.
diff --git a/magent2/environments/adversarial_pursuit/adversarial_pursuit.py b/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

diff --git a/magent2/environments/battle/battle.py b/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

diff --git a/magent2/environments/battlefield/battlefield.py b/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

diff --git a/magent2/environments/combined_arms/combined_arms.py b/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

diff --git a/magent2/environments/gather/gather.py b/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

diff --git a/magent2/environments/magent_env.py b/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

diff --git a/magent2/environments/tiger_deer/tiger_deer.py b/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