瀏覽代碼

Minor bugfix on FlatObsWrapper

Maxime Chevalier-Boisvert 7 年之前
父節點
當前提交
0ecec6bce9
共有 1 個文件被更改,包括 1 次插入1 次删除
  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