gryf/coach
1
0 Fork 0
mirror of https://github.com/gryf/coach.git synced 2026-10-09 15:22:36 +02:00
Code Issues Projects Releases Wiki Activity

fix for dumping movies, without rendering, for pendulum_with_goals

Browse Source
This commit is contained in:
Gal Leibovich authored and GitHub committed 2018-08-14 18:13:44 +03:00
1 parent e783157b15
commit ab5a81c7ee
1 file changed
+2 -1
Download Patch File Download Diff File Expand all files Collapse all files
rl_coach/environments/gym_environment.py
+2 -1
View File
@@ -425,7 +425,8 @@ class GymEnvironment(Environment):
:param camera_idx: The index of the camera to use. Should be defined in the model
:return: None
"""
if self.env.unwrapped.viewer.cam.fixedcamid != camera_idx and self.env.unwrapped.viewer._ncam > camera_idx:
if self.env.unwrapped.viewer is not None and self.env.unwrapped.viewer.cam.fixedcamid != camera_idx and\
self.env.unwrapped.viewer._ncam > camera_idx:
from mujoco_py.generated import const
self.env.unwrapped.viewer.cam.type = const.CAMERA_FIXED
self.env.unwrapped.viewer.cam.fixedcamid = camera_idx