Bläddra i källkod

Allow custom goal colours (#451)

Matt Barker 6 månader sedan
förälder
incheckning
a6cfc0e125
1 ändrade filer med 2 tillägg och 2 borttagningar
  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