robot out of view when using env.render('rgb_array')

[HareshKarnan]

Hi,

Thanks for this repo ! I'm having issues with the robot not being tracked properly by the camera. As a test, I trained a HalfCheetah agent and I visualize the learned policy by deploying it in the environment and visualizing it using env.render('rgb_array'). While the camera starts off tracking the robot well, after a few timesteps, the robot moves out of the frame. This is what it looks like :