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