浏览代码

fix Grid.rotate_left for non-square grids (#33)

A. Baisero 6 年之前
父节点
当前提交
02bc4b8cf8
共有 1 个文件被更改,包括 5 次插入5 次删除
  1. 5 5
      gym_minigrid/minigrid.py

+ 5 - 5
gym_minigrid/minigrid.py

@@ -469,12 +469,12 @@ class Grid:
         Rotate the grid to the left (counter-clockwise)
         """
 
-        grid = Grid(self.width, self.height)
+        grid = Grid(self.height, self.width)
 
-        for j in range(0, self.height):
-            for i in range(0, self.width):
-                v = self.get(self.width - 1 - j, i)
-                grid.set(i, j, v)
+        for i in range(self.width):
+            for j in range(self.height):
+                v = self.get(i, j)
+                grid.set(j, grid.height - 1 - i, v)
 
         return grid