by default assume state["observation"] is where the image for rendering can be found

2026-08-25 13:11:58 +02:00 · 2018-02-21 10:05:57 -05:00
parent f9f92a42fd
commit 0740ebcdac
1 changed files with 1 additions and 2 deletions
@@ -261,5 +261,4 @@ class EnvironmentWrapper(object):
        This can be different from the state. For example, mujoco's state is a measurements vector.
        :return: numpy array containing the image that will be rendered to the screen
        """
-        # TODO: probably needs revisiting
-        return self.state
+        return self.state['observation']