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

2025-12-17 11:10:20 +01:00 · 2018-02-20 11:45:35 -05:00
parent f9f92a42fd
commit 0740ebcdac
1 changed files with 1 additions and 2 deletions
--- a/environments/environment_wrapper.py
+++ b/environments/environment_wrapper.py
@@ -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']