next_obs -> obs

perezjln · May 19, 2024 · a8572fe · a8572fe
1 parent 8b234be
commit a8572fe
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 6 deletions.
diff --git a/gym_lowcostrobot/envs/lift_cube_env.py b/gym_lowcostrobot/envs/lift_cube_env.py
@@ -68,9 +68,9 @@ def step(self, action):
         terminated = object_z > self.threshold_distance
 
         # Return the next observation, reward, terminated flag, and additional info
-        next_observation = self.get_observation()
+        observation = self.get_observation()
 
         # Check if the episode is timed out, fill info dictionary
         info = self.get_info()
 
-        return next_observation, reward, terminated, False, info
+        return observation, reward, terminated, False, info
diff --git a/gym_lowcostrobot/envs/pick_place_cube_env.py b/gym_lowcostrobot/envs/pick_place_cube_env.py
@@ -77,9 +77,9 @@ def step(self, action):
         terminated = distance < self.threshold_distance
 
         # Return the next observation, reward, terminated flag, and additional info
-        next_observation = self.get_observation()
+        observation = self.get_observation()
 
         # Check if the episode is timed out, fill info dictionary
         info = self.get_info()
 
-        return next_observation, reward, terminated, False, info
+        return observation, reward, terminated, False, info
diff --git a/gym_lowcostrobot/envs/push_cube_env.py b/gym_lowcostrobot/envs/push_cube_env.py
@@ -75,9 +75,9 @@ def step(self, action):
         terminated = distance < self.threshold_distance
 
         # Return the next observation, reward, terminated flag, and additional info
-        next_observation = self.get_observation()
+        observation = self.get_observation()
 
         # Check if the episode is timed out, fill info dictionary
         info = self.get_info()
 
-        return next_observation, reward, terminated, False, info
+        return observation, reward, terminated, False, info