Update jumanji/environments/routing/pacman/env.py

Co-authored-by: Sasha <[email protected]>
instadeepai · Jan 8, 2024 · 8d19d78 · 8d19d78
1 parent 64687c9
commit 8d19d78
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/jumanji/environments/routing/pacman/env.py b/jumanji/environments/routing/pacman/env.py
@@ -444,8 +444,8 @@ def check_power_up(
         ps = jnp.array([ps_y, ps_x])
 
         # Check if player and power_up position are shared
-        valid = (ps == power_up_locations).all(axis=-1).any()
-        eat = 1 * valid
+        on_powerup = (ps == power_up_locations).all(axis=-1).any()
+        eat = on_powerup.astype(int)
         mask = (ps == power_up_locations).all(axis=-1)
         invert_mask = mask != True  # type: ignore # noqa: E712
         invert_mask = invert_mask.reshape(4, 1)