Addresses oxwhirl#72

douglasrizzo · Jul 9, 2021 · 8042fdc · 8042fdc
1 parent fcdd0d3
commit 8042fdc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/smac/env/starcraft2/starcraft2.py b/smac/env/starcraft2/starcraft2.py
@@ -742,7 +742,7 @@ def reward_battle(self):
                     delta_enemy += prev_health - e_unit.health - e_unit.shield
 
         if self.reward_only_positive:
-            reward = abs(delta_enemy + delta_deaths)  # shield regeneration
+            reward = max(0, delta_enemy + delta_deaths)  # shield regeneration
         else:
             reward = delta_enemy + delta_deaths - delta_ally