Allow custom goal colours (#451)

Farama-Foundation · Oct 6, 2024 · a6cfc0e · a6cfc0e
1 parent 715394f
commit a6cfc0e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/minigrid/core/world_object.py b/minigrid/core/world_object.py
@@ -107,8 +107,8 @@ def render(self, r: np.ndarray) -> np.ndarray:
 
 
 class Goal(WorldObj):
-    def __init__(self):
-        super().__init__("goal", "green")
+    def __init__(self, color: str = "green"):
+        super().__init__("goal", color)
 
     def can_overlap(self):
         return True