Farama-Foundation / Minigrid

Simple and easily configurable grid world environments for reinforcement learning
https://minigrid.farama.org/
Other
2.09k stars 604 forks source link

Bug Fix for the RGBImgPartialObsWrapper view size issue #421

Open vishwassathish opened 7 months ago

vishwassathish commented 7 months ago

Description

As reported in the bug, the RGBImgPartialObsWrapper ignored changes to agent_view_size parameter when rendering an RGB image. The issue was replicated and it was found that env.agent_view_size and env.unwrapped.agent_view_size had different values after applying a view size wrapper ViewSizeWrapper(env, agent_view_size=5).

Fixes # (issue)

Made a small change to the RGBImgPartialObsWrapper method in wrappers.py by adding an agent_view_size parameter. Also passed this argument to get_pov_render() and gen_obs_grid() functions in minigrid_env.py

Type of change

Please delete options that are not relevant.

Screenshots

Please attach before and after screenshots of the change if applicable. output_before output_after

Checklist: