Fix a bug in mask wrapper.

ikostrikov · May 7, 2019 · 8258f95 · 8258f95
1 parent b4133ec
commit 8258f95
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/a2c_ppo_acktr/envs.py b/a2c_ppo_acktr/envs.py
@@ -125,7 +125,7 @@ def reset(self, **kwargs):
 class MaskGoal(gym.ObservationWrapper):
     def observation(self, observation):
         if self.env._elapsed_steps > 0:
-            observation[-2:0] = 0
+            observation[-2:] = 0
         return observation