refactor: Improve code style

generals/agents/expander_agent.py

@@ -18,7 +18,7 @@ def play(self, observation):
 
         valid_actions = np.argwhere(mask == 1)
         if len(valid_actions) == 0:  # No valid actions
-            return (1, np.array([0, 0]), 0, 0)
+            return 1, np.array([0, 0]), 0, 0
 
         army = observation["army"]
         opponent = observation["opponent_cells"]

generals/agents/random_agent.py

@@ -20,7 +20,7 @@ def play(self, observation):
 
         valid_actions = np.argwhere(mask == 1)
         if len(valid_actions) == 0:  # No valid actions
-            return (1, (0, 0), 0, 0)
+            return 1, (0, 0), 0, 0
 
         pass_turn = 0 if np.random.rand() > self.idle_probability else 1
         split_army = 0 if np.random.rand() > self.split_probability else 1

generals/gui/rendering.py

@@ -110,7 +110,7 @@ def render_stats(self):
             )
             # add opacity to the color, where color is a tuple (r,g,b)
             if name in self.agent_fov and not self.agent_fov[name]:
-                color = tuple([int(0.5 * c) for c in color])
+                color = tuple([int(0.5 * rgb) for rgb in color])
             self.render_cell_text(self.score_cols["Agent"][i], name, bg_color=color)
 
         # Write other columns

generals/integrations/gymnasium_integration.py

@@ -26,6 +26,10 @@ def __init__(
         reward_fn: RewardFn = None,
         render_mode=None,
     ):
+        self.game = None
+        self.gui = None
+        self.replay = None
+
         self.render_mode = render_mode
         self.reward_fn = self.default_reward if reward_fn is None else reward_fn
         self.grid_factory = grid_factory

generals/integrations/pettingzoo_integration.py

@@ -30,6 +30,10 @@ def __init__(
         reward_fn: RewardFn = None,
         render_mode=None,
     ):
+        self.game = None
+        self.gui = None
+        self.replay = None
+
         self.render_mode = render_mode
         self.grid_factory = grid_factory