From 14671d7b08dd02fd693e1e294a8be738f7ea6b13 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Florian=20Schr=C3=B6der?= <florian.schroeder@uni-bielefeld.de> Date: Thu, 8 Feb 2024 20:36:04 +0100 Subject: [PATCH] Update overcooked_environment.py --- overcooked_simulator/overcooked_environment.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/overcooked_simulator/overcooked_environment.py b/overcooked_simulator/overcooked_environment.py index 00a5794c..847db60b 100644 --- a/overcooked_simulator/overcooked_environment.py +++ b/overcooked_simulator/overcooked_environment.py @@ -904,9 +904,9 @@ class Environment: name = "Meat" if name in self.vector_state_generation.meals: - idx = self.vector_state_generation.meals.index(name) + idx = 3 + self.vector_state_generation.meals.index(name) elif name in self.vector_state_generation.ingredients: - idx = len( + idx = 3 + len( self.vector_state_generation.meals ) + self.vector_state_generation.ingredients.index(name) else: -- GitLab