diff --git a/minigrid/core/world_object.py b/minigrid/core/world_object.py index a1fe01911..ea13f6d45 100644 --- 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