Maxime Chevalier-Boisvert 6 år sedan
förälder
incheckning
6e77fbef44
1 ändrade filer med 1 tillägg och 1 borttagningar
  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