浏览代码

Fixed issues with wrappers

Maxime Chevalier-Boisvert 7 年之前
父节点
当前提交
e7e870ce2d
共有 2 个文件被更改,包括 4 次插入1 次删除
  1. 3 0
      gym_minigrid/__init__.py
  2. 1 1
      gym_minigrid/wrappers.py

+ 3 - 0
gym_minigrid/__init__.py

@@ -1,2 +1,5 @@
 # Import the envs module so that envs register themselves
 import gym_minigrid.envs
+
+# Import wrappers so it's accessible when installing with pip
+import gym_minigrid.wrappers

+ 1 - 1
gym_minigrid/wrappers.py

@@ -106,7 +106,7 @@ class FlatObsWrapper(gym.core.ObservationWrapper):
             assert len(mission) <= self.maxStrLen, "mission string too long"
             mission = mission.lower()
 
-            strArray = np.zeros(shape=(self.maxStrLen, self.numCharCodes))
+            strArray = np.zeros(shape=(self.maxStrLen, self.numCharCodes), dtype='float32')
 
             for idx, ch in enumerate(mission):
                 if ch >= 'a' and ch <= 'z':