benelot / pybullet-gym

Open-source implementations of OpenAI Gym MuJoCo environments for use with the OpenAI Gym Reinforcement Learning Research Platform.
https://pybullet.org/
Other
823 stars 123 forks source link

Rendering (mode='rgb_array') error #56

Closed axelbr closed 3 years ago

axelbr commented 4 years ago

In the base_envs of mujoco and roboschool, the rgb_array is an one dimensional array. This leads to the following error: IndexError: too many indices for array. I can propose a PR with a fix, if appreciated.