Fix numpy deprecation warnings in pistonball/waterworld (#1157)

Farama-Foundation · Jan 21, 2024 · 161696c · 161696c
1 parent ba8dbe9
commit 161696c
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/pettingzoo/butterfly/pistonball/pistonball.py b/pettingzoo/butterfly/pistonball/pistonball.py
@@ -621,7 +621,8 @@ def step(self, action):
         action = np.asarray(action)
         agent = self.agent_selection
         if self.continuous:
-            self.move_piston(self.pistonList[self.agent_name_mapping[agent]], action)
+            # action is a 1 item numpy array, move_piston expects a scalar
+            self.move_piston(self.pistonList[self.agent_name_mapping[agent]], action[0])
         else:
             self.move_piston(
                 self.pistonList[self.agent_name_mapping[agent]], action - 1

diff --git a/pettingzoo/sisl/waterworld/waterworld_base.py b/pettingzoo/sisl/waterworld/waterworld_base.py
@@ -180,8 +180,8 @@ def add_obj(self):
                 Evaders(
                     x,
                     y,
-                    vx,
-                    vy,
+                    vx[0],
+                    vy[0],
                     radius=2 * self.base_radius,
                     collision_type=i + 1000,
                     max_speed=self.evader_speed,
@@ -198,8 +198,8 @@ def add_obj(self):
                 Poisons(
                     x,
                     y,
-                    vx,
-                    vy,
+                    vx[0],
+                    vy[0],
                     radius=0.75 * self.base_radius,
                     collision_type=i + 2000,
                     max_speed=self.poison_speed,