Maxime Chevalier-Boisvert hace 6 años
padre
commit
6e77fbef44
Se han modificado 1 ficheros con 1 adiciones y 1 borrados
  1. 1 1
      gym_minigrid/rendering.py

+ 1 - 1
gym_minigrid/rendering.py

@@ -144,7 +144,7 @@ class Renderer:
         numBytes = self.width * self.height * 3
         buf = self.img.bits().asstring(numBytes)
         output = np.frombuffer(buf, dtype='uint8')
-        output = output.reshape((height, width, 3))
+        output = output.reshape((self.height, self.width, 3))
 
         return output