Fix death penalty mario

Farama-Foundation · Jun 29, 2023 · 11015b1 · 11015b1
1 parent d98f068
commit 11015b1
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mo_gymnasium/envs/mario/mario.py b/mo_gymnasium/envs/mario/mario.py
@@ -162,8 +162,6 @@ def step(self, action):
         if "death" in self.objectives:
             vec_reward[obj_idx] = death_r
             obj_idx += 1
-        elif self.death_as_penalty:
-            vec_reward += death_r  # add death reward to all objectives
 
         # 4. coin
         coin_r = 0.0
@@ -182,6 +180,8 @@ def step(self, action):
             vec_reward[obj_idx] = enemy_r
             obj_idx += 1
 
+        if self.death_as_penalty:
+            vec_reward += death_r  # add death reward to all objectives
         ############################################################################
 
         if self.done_when_dead: