reformat

docs/code_examples/aec_rps.py

@@ -91,7 +91,9 @@ def render(self):
         up a graphical window, or open up some other display that a human can see and understand.
         """
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         if len(self.agents) == 2:

docs/code_examples/parallel_rps.py

@@ -86,7 +86,9 @@ def render(self):
         up a graphical window, or open up some other display that a human can see and understand.
         """
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         if len(self.agents) == 2:

pettingzoo/atari/base_atari_env.py

@@ -237,10 +237,14 @@ def step(self, action_dict):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
-        assert self.render_mode in self.metadata["render_modes"], f"{self.render_mode} is not a valid render mode"
+        assert (
+            self.render_mode in self.metadata["render_modes"]
+        ), f"{self.render_mode} is not a valid render mode"
         (screen_width, screen_height) = self.ale.getScreenDims()
         image = self.ale.getScreenRGB()
         if self.render_mode == "human":

pettingzoo/butterfly/cooperative_pong/cooperative_pong.py

@@ -187,7 +187,9 @@ def enable_render(self):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         if not self.renderOn and self.render_mode == "human":
@@ -199,7 +201,8 @@ def render(self):
             pygame.display.flip()
         return (
             np.transpose(observation, axes=(1, 0, 2))
-            if self.render_mode == "rgb_array" else None
+            if self.render_mode == "rgb_array"
+            else None
         )
 
     def observe(self):

pettingzoo/butterfly/knights_archers_zombies/knights_archers_zombies.py

@@ -75,7 +75,7 @@ def __init__(
             vector_state,
             use_typemasks,
             transformer,
-            render_mode
+            render_mode,
         )
         # variable state space
         self.transformer = transformer
@@ -612,7 +612,9 @@ def draw(self):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         if not self.render_on and self.render_mode == "human":
@@ -624,7 +626,8 @@ def render(self):
             pygame.display.flip()
         return (
             np.transpose(observation, axes=(1, 0, 2))
-            if self.render_mode == "rgb_array" else None
+            if self.render_mode == "rgb_array"
+            else None
         )
 
     def close(self):

pettingzoo/butterfly/pistonball/pistonball.py

@@ -62,7 +62,7 @@ def __init__(
         ball_friction=0.3,
         ball_elasticity=1.5,
         max_cycles=125,
-        render_mode=None
+        render_mode=None,
     ):
         EzPickle.__init__(
             self,
@@ -75,7 +75,7 @@ def __init__(
             ball_friction,
             ball_elasticity,
             max_cycles,
-            render_mode
+            render_mode,
         )
         self.dt = 1.0 / FPS
         self.n_pistons = n_pistons
@@ -513,7 +513,9 @@ def get_local_reward(self, prev_position, curr_position):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         if self.render_mode == "human" and not self.renderOn:
@@ -528,7 +530,8 @@ def render(self):
             pygame.display.flip()
         return (
             np.transpose(observation, axes=(1, 0, 2))
-            if self.render_mode == "rgb_array" else None
+            if self.render_mode == "rgb_array"
+            else None
         )
 
     def step(self, action):

pettingzoo/classic/chess/chess.py

@@ -1,6 +1,5 @@
-import gym
-
 import chess
+import gym
 import numpy as np
 from gym import spaces
 
@@ -152,7 +151,9 @@ def step(self, action):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
         else:
             print(self.board)
 

pettingzoo/classic/connect_four/connect_four.py

@@ -157,7 +157,9 @@ def reset(self, seed=None, return_info=False, options=None):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         screen_width = 1287
@@ -217,7 +219,8 @@ def render(self):
 
         return (
             np.transpose(observation, axes=(1, 0, 2))
-            if self.render_mode == "rgb_array" else None
+            if self.render_mode == "rgb_array"
+            else None
         )
 
     def close(self):

pettingzoo/classic/go/go.py

@@ -42,7 +42,9 @@ class raw_env(AECEnv):
         "render_fps": 2,
     }
 
-    def __init__(self, board_size: int = 19, komi: float = 7.5, render_mode: Optional[str] = None):
+    def __init__(
+        self, board_size: int = 19, komi: float = 7.5, render_mode: Optional[str] = None
+    ):
         # 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__()
@@ -218,7 +220,9 @@ def reset(self, seed=None, return_info=False, options=None):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         screen_width = 1026
@@ -309,7 +313,8 @@ def render(self):
 
         return (
             np.transpose(observation, axes=(1, 0, 2))
-            if self.render_mode == "rgb_array" else None
+            if self.render_mode == "rgb_array"
+            else None
         )
 
     def close(self):

pettingzoo/classic/hanabi/hanabi.py

@@ -76,7 +76,7 @@ def __init__(
         max_life_tokens: int = 3,
         observation_type: int = 1,
         random_start_player: bool = False,
-        render_mode: Optional[str] = None
+        render_mode: Optional[str] = None,
     ):
         """Initializes the `raw_env` class.
 
@@ -133,7 +133,7 @@ def __init__(
             max_life_tokens,
             observation_type,
             random_start_player,
-            render_mode
+            render_mode,
         )
 
         # ToDo: Starts
@@ -392,7 +392,9 @@ def render(self):
         Supports console print only.
         """
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         player_data = self.latest_observations["player_observations"]

pettingzoo/classic/rlcard_envs/gin_rummy.py

@@ -35,7 +35,7 @@ def __init__(
         knock_reward: float = 0.5,
         gin_reward: float = 1.0,
         opponents_hand_visible=False,
-        render_mode=None
+        render_mode=None,
     ):
         EzPickle.__init__(self, knock_reward, gin_reward, render_mode)
         self._opponents_hand_visible = opponents_hand_visible
@@ -84,7 +84,9 @@ def observe(self, agent):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         for player in self.possible_agents:

pettingzoo/classic/rlcard_envs/leduc_holdem.py

@@ -30,7 +30,9 @@ def __init__(self, num_players=2, render_mode=None):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         for player in self.possible_agents:

pettingzoo/classic/rlcard_envs/texas_holdem.py

@@ -50,7 +50,9 @@ def __init__(self, num_players=2, render_mode=None):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         def calculate_width(self, screen_width, i):
@@ -286,5 +288,7 @@ def calculate_height(screen_height, divisor, multiplier, tile_size, offset):
         observation = np.array(pygame.surfarray.pixels3d(self.screen))
 
         return (
-            np.transpose(observation, axes=(1, 0, 2)) if self.render_mode == "rgb_array" else None
+            np.transpose(observation, axes=(1, 0, 2))
+            if self.render_mode == "rgb_array"
+            else None
         )

pettingzoo/classic/rlcard_envs/texas_holdem_no_limit.py

@@ -76,7 +76,9 @@ def __init__(self, num_players=2, render_mode=None):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         def calculate_width(self, screen_width, i):
@@ -312,5 +314,7 @@ def calculate_height(screen_height, divisor, multiplier, tile_size, offset):
         observation = np.array(pygame.surfarray.pixels3d(self.screen))
 
         return (
-            np.transpose(observation, axes=(1, 0, 2)) if self.render_mode == "rgb_array" else None
+            np.transpose(observation, axes=(1, 0, 2))
+            if self.render_mode == "rgb_array"
+            else None
         )

pettingzoo/classic/rps/rps.py

@@ -108,7 +108,9 @@ def reinit(self):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         def offset(i, size, offset=0):
@@ -309,7 +311,8 @@ def offset(i, size, offset=0):
 
         return (
             np.transpose(observation, axes=(1, 0, 2))
-            if self.render_mode == "rgb_array" else None
+            if self.render_mode == "rgb_array"
+            else None
         )
 
     def observe(self, agent):

pettingzoo/classic/tictactoe/tictactoe.py

@@ -1,5 +1,5 @@
-import numpy as np
 import gym
+import numpy as np
 from gym import spaces
 
 from pettingzoo import AECEnv
@@ -149,8 +149,11 @@ def reset(self, seed=None, return_info=False, options=None):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
+
         def getSymbol(input):
             if input == 0:
                 return "-"

pettingzoo/magent/magent_env.py

@@ -29,7 +29,7 @@ def __init__(
         reward_range,
         minimap_mode,
         extra_features,
-        render_mode=None
+        render_mode=None,
     ):
         self.map_size = map_size
         self.max_cycles = max_cycles
@@ -141,7 +141,9 @@ def _calc_state_shape(self):
 
     def render(self):
         if self.render_mode is None:
-            gym.logger.WARN("You are calling render method without specifying any render mode.")
+            gym.logger.WARN(
+                "You are calling render method without specifying any render mode."
+            )
             return
 
         if self._renderer is None: