Explorar o código

Minor bugfix on FlatObsWrapper

Maxime Chevalier-Boisvert %!s(int64=7) %!d(string=hai) anos
pai
achega
0ecec6bce9
Modificáronse 1 ficheiros con 1 adicións e 1 borrados
  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