Kaynağa Gözat

Minor bugfix on FlatObsWrapper

Maxime Chevalier-Boisvert 7 yıl önce
ebeveyn
işleme
0ecec6bce9
1 değiştirilmiş dosya ile 1 ekleme ve 1 silme
  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