From 04bf8fbdf0d75c684d2d7a6070afb5907e7c27e0 Mon Sep 17 00:00:00 2001
From: helpingstar <iamhelpingstar@gmail.com>
Date: Tue, 29 Oct 2024 00:17:55 +0900
Subject: [PATCH] Refactor: Minor code style improvements (#246)

---
 jumanji/environments/routing/pac_man/viewer.py | 6 +-----
 1 file changed, 1 insertion(+), 5 deletions(-)

diff --git a/jumanji/environments/routing/pac_man/viewer.py b/jumanji/environments/routing/pac_man/viewer.py
index 064293152..bf9ed5174 100644
--- a/jumanji/environments/routing/pac_man/viewer.py
+++ b/jumanji/environments/routing/pac_man/viewer.py
@@ -65,10 +65,7 @@ def render(self, state: Union[Observation, State]) -> Optional[NDArray]:
             RGB array if the render_mode is RenderMode.RGB_ARRAY.
         """
         self._clear_display()
-        (
-            fig,
-            ax,
-        ) = self._get_fig_ax()
+        fig, ax = self._get_fig_ax()
         ax.clear()
         fig.suptitle(f"PacMan    Score: {int(state.score)}", size=15)
         self._add_grid_image(state, ax)
@@ -350,6 +347,5 @@ def set_ghost_colours_scared_details(
 
     obs = [layer_1, layer_2, layer_3]
     rgb = jnp.stack(obs, axis=-1)
-    expand_rgb
 
     return rgb