Refactor: Minor code style improvements (#246)

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