chore: remove extra lmabda function

instadeepai · Jan 8, 2024 · 8b03169 · 8b03169
1 parent 73a35d9
commit 8b03169
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/jumanji/environments/routing/pacman/utils.py b/jumanji/environments/routing/pacman/utils.py
@@ -370,11 +370,8 @@ def check_collisions(
         old_ghost_pos: chex.Array,
     ) -> Tuple[chex.Array, chex.Numeric, chex.Numeric, chex.Numeric, chex.Numeric]:
         """Check if ghost has collided with player"""
-        eat = lambda: True
-        no_eat = lambda: False
         frightened_time = state.frightened_state_time
-
-        is_eat = jax.lax.cond(frightened_time > 0, eat, no_eat)
+        is_eat = frightened_time > 0
 
         ghost_p = Position(y=ghost_pos[0], x=ghost_pos[1])
         old_ghost_p = Position(y=old_ghost_pos[0], x=old_ghost_pos[1])