Fix warnings from waterworld test

These came because of numpy changes that didn't like using a numpy array of 1 item as a scalar.
Farama-Foundation · Jan 21, 2024 · d1b3638 · d1b3638
1 parent bc62f82
commit d1b3638
Showing 1 changed file with 4 additions and 4 deletions.
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,