Added EzPickle support to all envs

pettingzoo/classic/chess/chess.py

@@ -90,6 +90,7 @@
 import numpy as np
 from gymnasium import spaces
 from gymnasium.error import DependencyNotInstalled
+from gymnasium.utils import EzPickle
 
 from pettingzoo import AECEnv
 from pettingzoo.utils import wrappers
@@ -106,7 +107,7 @@ def env(render_mode=None):
     return env
 
 
-class raw_env(AECEnv):
+class raw_env(AECEnv, EzPickle):
     metadata = {
         "render_modes": ["human", "ansi", "rgb_array"],
         "name": "chess_v5",
@@ -115,6 +116,7 @@ class raw_env(AECEnv):
     }
 
     def __init__(self, render_mode=None):
+        EzPickle.__init__(self, render_mode)
         super().__init__()
 
         self.board = chess.Board()

pettingzoo/classic/connect_four/connect_four.py

@@ -64,6 +64,7 @@
 import numpy as np
 import pygame
 from gymnasium import spaces
+from gymnasium.utils import EzPickle
 
 from pettingzoo import AECEnv
 from pettingzoo.utils import wrappers
@@ -90,7 +91,7 @@ def env(render_mode=None):
     return env
 
 
-class raw_env(AECEnv):
+class raw_env(AECEnv, EzPickle):
     metadata = {
         "render_modes": ["human", "rgb_array"],
         "name": "connect_four_v3",
@@ -99,6 +100,7 @@ class raw_env(AECEnv):
     }
 
     def __init__(self, render_mode=None):
+        EzPickle.__init__(self, render_mode)
         super().__init__()
         # 6 rows x 7 columns
         # blank space = 0

pettingzoo/classic/go/go.py

@@ -114,6 +114,7 @@
 import numpy as np
 import pygame
 from gymnasium import spaces
+from gymnasium.utils import EzPickle
 
 from pettingzoo import AECEnv
 from pettingzoo.utils import wrappers
@@ -142,7 +143,7 @@ def env(**kwargs):
     return env
 
 
-class raw_env(AECEnv):
+class raw_env(AECEnv, EzPickle):
     metadata = {
         "render_modes": ["human", "rgb_array"],
         "name": "go_v5",
@@ -153,6 +154,7 @@ class raw_env(AECEnv):
     def __init__(
         self, board_size: int = 19, komi: float = 7.5, render_mode: Optional[str] = None
     ):
+        EzPickle.__init__(self, board_size, komi, render_mode)
         # board_size: a int, representing the board size (board has a board_size x board_size shape)
         # komi: a float, representing points given to the second player.
         super().__init__()

pettingzoo/classic/rlcard_envs/leduc_holdem.py

@@ -82,6 +82,7 @@
 """
 
 import gymnasium
+from gymnasium.utils import EzPickle
 from rlcard.utils.utils import print_card
 
 from pettingzoo.utils import wrappers
@@ -103,7 +104,7 @@ def env(**kwargs):
     return env
 
 
-class raw_env(RLCardBase):
+class raw_env(RLCardBase, EzPickle):
     metadata = {
         "render_modes": ["human"],
         "name": "leduc_holdem_v4",
@@ -112,6 +113,7 @@ class raw_env(RLCardBase):
     }
 
     def __init__(self, num_players=2, render_mode=None):
+        EzPickle.__init__(self, num_players, render_mode)
         super().__init__("leduc-holdem", num_players, (36,))
         self.render_mode = render_mode
 

pettingzoo/classic/rlcard_envs/rlcard_base.py

@@ -1,12 +1,14 @@
 import numpy as np
 import rlcard
 from gymnasium import spaces
+from gymnasium.utils import EzPickle
 
 from pettingzoo import AECEnv
 
 
-class RLCardBase(AECEnv):
+class RLCardBase(AECEnv, EzPickle):
     def __init__(self, name, num_players, obs_shape):
+        EzPickle.__init__(self, name, num_players, obs_shape)
         super().__init__()
         self.name = name
         self.num_players = num_players

pettingzoo/classic/rlcard_envs/texas_holdem.py

@@ -82,6 +82,7 @@
 import gymnasium
 import numpy as np
 import pygame
+from gymnasium.utils import EzPickle
 
 from pettingzoo.utils import wrappers
 
@@ -114,7 +115,7 @@ def env(**kwargs):
     return env
 
 
-class raw_env(RLCardBase):
+class raw_env(RLCardBase, EzPickle):
     metadata = {
         "render_modes": ["human", "rgb_array"],
         "name": "texas_holdem_v4",
@@ -123,6 +124,7 @@ class raw_env(RLCardBase):
     }
 
     def __init__(self, num_players=2, render_mode=None):
+        EzPickle.__init__(self, num_players, render_mode)
         super().__init__("limit-holdem", num_players, (72,))
         self.render_mode = render_mode
 

pettingzoo/classic/rlcard_envs/texas_holdem_no_limit.py

@@ -95,6 +95,7 @@
 import numpy as np
 import pygame
 from gymnasium import spaces
+from gymnasium.utils import EzPickle
 
 from pettingzoo.utils import wrappers
 
@@ -127,7 +128,7 @@ def env(**kwargs):
     return env
 
 
-class raw_env(RLCardBase):
+class raw_env(RLCardBase, EzPickle):
     metadata = {
         "render_modes": ["human", "rgb_array"],
         "name": "texas_holdem_no_limit_v6",
@@ -136,6 +137,7 @@ class raw_env(RLCardBase):
     }
 
     def __init__(self, num_players=2, render_mode=None):
+        EzPickle.__init__(self, num_players, render_mode)
         super().__init__("no-limit-holdem", num_players, (54,))
         self.observation_spaces = self._convert_to_dict(
             [

pettingzoo/classic/rps/rps.py

@@ -117,6 +117,7 @@
 import numpy as np
 import pygame
 from gymnasium.spaces import Discrete
+from gymnasium.utils import EzPickle
 
 from pettingzoo import AECEnv
 from pettingzoo.utils import agent_selector, wrappers
@@ -153,7 +154,7 @@ def env(**kwargs):
 parallel_env = parallel_wrapper_fn(env)
 
 
-class raw_env(AECEnv):
+class raw_env(AECEnv, EzPickle):
     """Two-player environment for rock paper scissors.
 
     Expandable environment to rock paper scissors lizard spock action_6 action_7 ...
@@ -168,6 +169,7 @@ class raw_env(AECEnv):
     }
 
     def __init__(self, num_actions=3, max_cycles=15, render_mode=None):
+        EzPickle.__init__(self, num_actions, max_cycles, render_mode)
         self.max_cycles = max_cycles
 
         # number of actions must be odd and greater than 3

pettingzoo/classic/tictactoe/tictactoe.py

@@ -72,6 +72,7 @@
 import gymnasium
 import numpy as np
 from gymnasium import spaces
+from gymnasium.utils import EzPickle
 
 from pettingzoo import AECEnv
 from pettingzoo.utils import agent_selector, wrappers
@@ -90,7 +91,7 @@ def env(render_mode=None):
     return env
 
 
-class raw_env(AECEnv):
+class raw_env(AECEnv, EzPickle):
     metadata = {
         "render_modes": ["human"],
         "name": "tictactoe_v3",
@@ -99,6 +100,7 @@ class raw_env(AECEnv):
     }
 
     def __init__(self, render_mode=None):
+        EzPickle.__init__(self, render_mode)
         super().__init__()
         self.board = Board()
 

pettingzoo/mpe/simple/simple.py

@@ -43,6 +43,7 @@
 """
 
 import numpy as np
+from gymnasium.utils import EzPickle
 
 from pettingzoo.utils.conversions import parallel_wrapper_fn
 
@@ -51,8 +52,9 @@
 from .._mpe_utils.simple_env import SimpleEnv, make_env
 
 
-class raw_env(SimpleEnv):
+class raw_env(SimpleEnv, EzPickle):
     def __init__(self, max_cycles=25, continuous_actions=False, render_mode=None):
+        EzPickle.__init__(self, max_cycles, continuous_actions, render_mode)
         scenario = Scenario()
         world = scenario.make_world()
         super().__init__(

pettingzoo/sisl/waterworld/waterworld.py

@@ -136,6 +136,8 @@
 
 """
 
+from gymnasium.utils import EzPickle
+
 from pettingzoo import AECEnv
 from pettingzoo.utils import agent_selector, wrappers
 from pettingzoo.utils.conversions import parallel_wrapper_fn
@@ -154,7 +156,7 @@ def env(**kwargs):
 parallel_env = parallel_wrapper_fn(env)
 
 
-class raw_env(AECEnv):
+class raw_env(AECEnv, EzPickle):
     metadata = {
         "render_modes": ["human", "rgb_array"],
         "name": "waterworld_v4",
@@ -163,6 +165,7 @@ class raw_env(AECEnv):
     }
 
     def __init__(self, *args, **kwargs):
+        EzPickle.__init__(self, *args, **kwargs)
         super().__init__()
         self.env = _env(*args, **kwargs)