Add env infos for custom hopper and walker env

rail-berkeley · Jan 1, 2019 · 94f7809 · 94f7809
1 parent c2419b6
commit 94f7809
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 2 deletions.
diff --git a/softlearning/environments/gym/mujoco/hopper_env.py b/softlearning/environments/gym/mujoco/hopper_env.py
@@ -113,7 +113,10 @@ def step(self, action):
         observation = self._get_obs()
         reward = rewards - costs
         done = self.done
-        info = {}
+        info = {
+            'x_position': x_position_after,
+            'x_velocity': x_velocity,
+        }
 
         return observation, reward, done, info
 

diff --git a/softlearning/environments/gym/mujoco/walker2d_env.py b/softlearning/environments/gym/mujoco/walker2d_env.py
@@ -105,7 +105,10 @@ def step(self, action):
         observation = self._get_obs()
         reward = rewards - costs
         done = self.done
-        info = {}
+        info = {
+            'x_position': x_position_after,
+            'x_velocity': x_velocity,
+        }
 
         return observation, reward, done, info