Add x position to Ant observation

tristandeleu · Oct 13, 2018 · b1cd6f1 · b1cd6f1
1 parent b9a84c7
commit b1cd6f1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/maml_rl/envs/mujoco/ant.py b/maml_rl/envs/mujoco/ant.py
@@ -13,7 +13,7 @@ def action_scaling(self):
 
     def _get_obs(self):
         return np.concatenate([
-            self.sim.data.qpos.flat[2:],
+            self.sim.data.qpos.flat,
             self.sim.data.qvel.flat,
             np.clip(self.sim.data.cfrc_ext, -1, 1).flat,
             self.sim.data.get_body_xmat("torso").flat,