Update pusher_v5.py

Farama-Foundation · Dec 9, 2023 · 5ac1bb2 · 5ac1bb2
1 parent b1f7fb4
commit 5ac1bb2
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/gymnasium/envs/mujoco/pusher_v5.py b/gymnasium/envs/mujoco/pusher_v5.py
@@ -221,13 +221,14 @@ def __init__(
         }
 
     def step(self, action):
-        reward, reward_info = self._get_rew(action)
         self.do_simulation(action, self.frame_skip)
 
         observation = self._get_obs()
+        reward, reward_info = self._get_rew(action)
         info = reward_info
         if self.render_mode == "human":
             self.render()
+
         return observation, reward, False, False, info
 
     def _get_rew(self, action):