Bläddra i källkod

Minor bugfix on FlatObsWrapper

Maxime Chevalier-Boisvert 7 år sedan
förälder
incheckning
0ecec6bce9
1 ändrade filer med 1 tillägg och 1 borttagningar
  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