type hints

overcooked_simulator/counters.py

@@ -145,7 +145,7 @@ class ServingWindow(Counter):
         self,
         pos,
         game_score: GameScore,
-        meals: list[str],
+        meals: set[str],
         plate_dispenser: PlateDispenser = None,
     ):
         self.game_score = game_score

overcooked_simulator/game_items.py

@@ -77,7 +77,7 @@ class Item:
     def extra_repr(self):
         return ""
 
-    def can_combine(self, other):
+    def can_combine(self, other) -> bool:
         return False
 
     def combine(self, other) -> Item | None:
@@ -106,7 +106,6 @@ class CookingEquipment(Item):
         super().__init__(*args, **kwargs)
         self.transitions = transitions
         self.active_transition: Optional[dict] = None
-        # self.content: Item | list[Item] = []  # list if cooking, meal item when done
 
         self.content_ready: Item | None = None
         self.content_list: list[Item] = []
@@ -116,7 +115,7 @@ class CookingEquipment(Item):
         for transition in self.transitions.values():
             transition["recipe"] = collections.Counter(transition["needs"])
 
-    def can_combine(self, other):
+    def can_combine(self, other) -> bool:
         # already cooking or nothing to combine
         if other is None:
             return False
@@ -134,7 +133,7 @@ class CookingEquipment(Item):
             ingredients <= recipe["recipe"] for recipe in self.transitions.values()
         )
 
-    def combine(self, other):
+    def combine(self, other) -> Item | None:
         return_value = None
         if isinstance(other, CookingEquipment):
             self.content_list.extend(other.content_list)
@@ -199,7 +198,7 @@ class CookingEquipment(Item):
 
 
 class Plate(CookingEquipment):
-    def __init__(self, transitions, clean, content=None, *args, **kwargs):
+    def __init__(self, transitions, clean, *args, **kwargs):
         self.clean = clean
         super().__init__(
             name=self.create_name(), transitions=transitions, *args, **kwargs