add unwrapped to wrapper works

Author: BartekCupial

balrog/environments/nle/base.py

@@ -78,13 +78,13 @@ def nle_obsv_type(self, nle_obsv):
 
     def render(self, mode="human"):
         if mode == "tiles":
-            obs = self.env.last_observation
-            glyphs = obs[self.env._observation_keys.index("glyphs")]
+            obs = self.env.unwrapped.last_observation
+            glyphs = obs[self.env.unwrapped._observation_keys.index("glyphs")]
             return rgb_render_image(glyphs)
         elif mode == "tty_image":
-            obs = self.env.last_observation
-            tty_chars = obs[self.env._observation_keys.index("tty_chars")]
-            tty_colors = obs[self.env._observation_keys.index("tty_colors")]
+            obs = self.env.unwrapped.last_observation
+            tty_chars = obs[self.env.unwrapped._observation_keys.index("tty_chars")]
+            tty_colors = obs[self.env.unwrapped._observation_keys.index("tty_colors")]
             return tty_render_image(tty_chars, tty_colors)
         else:
             return super().render(mode)