Maxime Chevalier-Boisvert %!s(int64=6) %!d(string=hai) anos
pai
achega
6e77fbef44
Modificáronse 1 ficheiros con 1 adicións e 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