From c5ed4702f19457a963c8cf7c53e905ad944a4e08 Mon Sep 17 00:00:00 2001
From: fheinrich <fheinrich@techfak.uni-bielefeld.de>
Date: Wed, 17 Jan 2024 17:55:45 +0100
Subject: [PATCH] Fixed putting multiple onions on a plate
---
overcooked_simulator/game_items.py | 10 ++++++++--
1 file changed, 8 insertions(+), 2 deletions(-)
diff --git a/overcooked_simulator/game_items.py b/overcooked_simulator/game_items.py
index cb674e41..1969e9b2 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
--
GitLab