From 25947d5c5c2e04aeb3335224bccab065d2e838be Mon Sep 17 00:00:00 2001 From: fheinrich <fheinrich@techfak.uni-bielefeld.de> Date: Fri, 15 Dec 2023 15:39:17 +0100 Subject: [PATCH] Simpler putting things on plate dispenser and picking new plates up --- overcooked_simulator/counters.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) diff --git a/overcooked_simulator/counters.py b/overcooked_simulator/counters.py index 3f2c5382..dfe0b24e 100644 --- a/overcooked_simulator/counters.py +++ b/overcooked_simulator/counters.py @@ -153,12 +153,9 @@ class Dispenser(Counter): ) def pick_up(self, on_hands: bool = True): - new_dispensing = self.dispensing.create_item() - if self.occupied_by != new_dispensing: - old_dispensing = self.occupied_by - self.occupied_by = new_dispensing - return old_dispensing - return new_dispensing + return_this = self.occupied_by + self.occupied_by = self.dispensing.create_item() + return return_this def drop_off(self, item: Item) -> Item | None: if self.occupied_by.can_combine(item): -- GitLab