Преглед на файлове

Remove extra dimension in FlatObsWrapper. In response to #86.

Maxime Chevalier-Boisvert преди 5 години
родител
ревизия
67bde16e38
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      gym_minigrid/wrappers.py

+ 1 - 1
gym_minigrid/wrappers.py

@@ -265,7 +265,7 @@ class FlatObsWrapper(gym.core.ObservationWrapper):
         self.observation_space = spaces.Box(
             low=0,
             high=255,
-            shape=(1, imgSize + self.numCharCodes * self.maxStrLen),
+            shape=(imgSize + self.numCharCodes * self.maxStrLen,),
             dtype='uint8'
         )