diff --git a/overcooked_simulator/counters.py b/overcooked_simulator/counters.py index 8e0722de59f62616374d1a455201954a3a7def00..999c2a6222093b688cbb0b63d2f3a4f5e49624b1 100644 --- a/overcooked_simulator/counters.py +++ b/overcooked_simulator/counters.py @@ -1,4 +1,9 @@ -from typing import Optional +from __future__ import annotations + +from typing import TYPE_CHECKING, Optional + +if TYPE_CHECKING: + from overcooked_simulator.overcooked_environment import GameScore import numpy as np import numpy.typing as npt @@ -101,7 +106,14 @@ class CuttingBoard(Counter): class ServingWindow(Counter): + def __init__(self, pos, gamescore: GameScore): + self.game_score = gamescore + super().__init__(pos) + def drop_off(self, item) -> HoldableItem | None: + reward = 5 + # TODO define rewards + self.game_score.increment_score(reward) return None def can_score(self, item):