Update jumanji/environments/logic/sliding_tile_puzzle/env.py

Co-authored-by: Sasha <[email protected]>

jumanji/environments/logic/sliding_tile_puzzle/env.py

@@ -152,7 +152,7 @@ def _move_empty_tile(
         puzzle: chex.Array,
         empty_tile_position: chex.Array,
         action: chex.Array,
-    ) -> Tuple[chex.Array, chex.Array]:
+    ) -> chex.Array:
         """Moves the empty tile in the given direction and returns the updated puzzle and reward."""
 
         # Compute the new position