gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 11:10:20 +01: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
2018-08-14 18:13:44 +03:00
committed by GitHub
parent e783157b15
commit ab5a81c7ee
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
rl_coach/environments/gym_environment.py
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 :param camera_idx: The index of the camera to use. Should be defined in the model
:return: None :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 from mujoco_py.generated import const
self.env.unwrapped.viewer.cam.type = const.CAMERA_FIXED self.env.unwrapped.viewer.cam.type = const.CAMERA_FIXED
self.env.unwrapped.viewer.cam.fixedcamid = camera_idx self.env.unwrapped.viewer.cam.fixedcamid = camera_idx