Преглед на файлове

Allow custom goal colours (#451)

Matt Barker преди 6 месеца
родител
ревизия
a6cfc0e125
променени са 1 файла, в които са добавени 2 реда и са изтрити 2 реда
  1. 2 2
      minigrid/core/world_object.py

+ 2 - 2
minigrid/core/world_object.py

@@ -107,8 +107,8 @@ class WorldObj:
 
 
 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