diff --git a/overcooked_simulator/counters.py b/overcooked_simulator/counters.py index 3f68bae69948eba8f059f73e68bbec1554f94c90..a66ae491ed90b227005331adcb0062c75a8bb10e 100644 --- a/overcooked_simulator/counters.py +++ b/overcooked_simulator/counters.py @@ -184,6 +184,12 @@ class Trash(Counter): class Stove(Counter): + def can_drop_off(self, item: Item) -> bool: + if self.occupied_by is None: + return isinstance(item, CookingEquipment) and item.name in ["Pot", "Pan"] + else: + return self.occupied_by.can_combine(item) + def progress(self): """Called by environment step function for time progression""" if (