defaulting render to tracking camera

zuoxingdong · Apr 15, 2019 · 1fa3b8f · 1fa3b8f
1 parent ec12ed0
commit 1fa3b8f
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dm2gym/dm_control_env.py b/dm2gym/dm_control_env.py
@@ -56,6 +56,8 @@ def reset(self):
         return timestep.observation
 
     def render(self, mode='human', **kwargs):
+        if 'camera_id' not in kwargs:
+            kwargs['camera_id'] = 0  # tracking camera
         img = self.env.physics.render(**kwargs)
         if mode == 'rgb_array':
             return img