Browse Source

Minor bugfix on FlatObsWrapper

Maxime Chevalier-Boisvert 7 years ago
parent
commit
0ecec6bce9
1 changed files with 1 additions and 1 deletions
  1. 1 1
      gym_minigrid/wrappers.py

+ 1 - 1
gym_minigrid/wrappers.py

@@ -119,6 +119,6 @@ class FlatObsWrapper(gym.core.ObservationWrapper):
             self.cachedStr = mission
             self.cachedArray = strArray
 
-        obs = np.hstack((image.flatten(), self.cachedArray.flatten()))
+        obs = np.concatenate((image.flatten(), self.cachedArray.flatten()))
 
         return obs