refactor(hopper_v4): add forward/health reward and ctrl cost to step …

…info
Farama-Foundation · Jul 12, 2023 · a971fa1 · a971fa1
1 parent ee067c7
commit a971fa1
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/gymnasium/envs/mujoco/hopper_v4.py b/gymnasium/envs/mujoco/hopper_v4.py
@@ -273,6 +273,9 @@ def step(self, action):
         reward = rewards - costs
         terminated = self.terminated
         info = {
+            "reward_forward": forward_reward,
+            "reward_ctrl": -ctrl_cost,
+            "reward_survive": healthy_reward,
             "x_position": x_position_after,
             "x_velocity": x_velocity,
         }