From eff4b8b7b80e6b218b7f992d909b152fbee51e71 Mon Sep 17 00:00:00 2001 From: Elshadai Tegegn <54620095+ElshadaiK@users.noreply.github.com> Date: Tue, 16 Jan 2024 11:51:08 -0600 Subject: [PATCH] Update jumanji/environments/logic/sliding_tile_puzzle/env.py Co-authored-by: Sasha --- jumanji/environments/logic/sliding_tile_puzzle/env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/jumanji/environments/logic/sliding_tile_puzzle/env.py b/jumanji/environments/logic/sliding_tile_puzzle/env.py index 055e87310..517bd515b 100644 --- a/jumanji/environments/logic/sliding_tile_puzzle/env.py +++ b/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