diff --git a/overcooked_simulator/game_items.py b/overcooked_simulator/game_items.py index cb674e41e05470e04fdd2ea38736fe50f9ed35a4..1969e9b25935664c6c201a99fde0937d7f4204ee 100644 --- a/overcooked_simulator/game_items.py +++ b/overcooked_simulator/game_items.py @@ -200,8 +200,14 @@ class CookingEquipment(Item): class Plate(CookingEquipment): def __init__(self, transitions, clean, *args, **kwargs): self.clean = clean + self.meals = set(transitions.keys()) super().__init__( - name=self.create_name(), transitions=transitions, *args, **kwargs + name=self.create_name(), + transitions={ + k: v for k, v in transitions.items() if not v["info"].equipment + }, + *args, + **kwargs, ) def finished_call(self): @@ -222,7 +228,7 @@ class Plate(CookingEquipment): and not self.content_list and self.clean ): - return other.content_list[0].name in self.transitions + return other.content_list[0].name in self.meals return False else: return True