Browse Source

Added ability to have colored walls, for colored rooms

Maxime Chevalier-Boisvert 7 years ago
parent
commit
929bcab4d0
1 changed files with 2 additions and 2 deletions
  1. 2 2
      gym_minigrid/envs/minigrid.py

+ 2 - 2
gym_minigrid/envs/minigrid.py

@@ -98,8 +98,8 @@ class Goal(WorldObj):
         ])
 
 class Wall(WorldObj):
-    def __init__(self):
-        super(Wall, self).__init__('wall', 'grey')
+    def __init__(self, color='grey'):
+        super(Wall, self).__init__('wall', color)
 
     def render(self, r):
         self._setColor(r)