From 92e401e7182c30a5566967cb982f491aadf8b73c Mon Sep 17 00:00:00 2001
From: fheinrich <fheinrich@techfak.uni-bielefeld.de>
Date: Fri, 15 Dec 2023 11:48:28 +0100
Subject: [PATCH] Only pot and pan can be put on stove
---
overcooked_simulator/counters.py | 6 ++++++
1 file changed, 6 insertions(+)
diff --git a/overcooked_simulator/counters.py b/overcooked_simulator/counters.py
index 3f68bae6..a66ae491 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 (
--
GitLab