From 0740ebcdac28a60360662a16b58b376a2377a4d5 Mon Sep 17 00:00:00 2001
From: Zach Dwiel <zach.dwiel@intel.com>
Date: Tue, 20 Feb 2018 11:45:35 -0500
Subject: [PATCH] by default assume state["observation"] is where the image for
 rendering can be found

---
 environments/environment_wrapper.py | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/environments/environment_wrapper.py b/environments/environment_wrapper.py
index 48fd0d8..d077c39 100644
--- 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']