fix: viewer

instadeepai · Mar 13, 2024 · 1bec548 · 1bec548
1 parent 8750260
commit 1bec548
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/jumanji/environments/logic/sliding_tile_puzzle/viewer.py b/jumanji/environments/logic/sliding_tile_puzzle/viewer.py
@@ -86,7 +86,7 @@ def animate(
 
         def make_frame(state_index: int) -> None:
             state = states[state_index]
-            self.draw_puzzle(ax, state.puzzle)
+            self.draw_puzzle(ax, state)
 
         self._animation = matplotlib.animation.FuncAnimation(
             fig,