fixed reward

Author: marcocspc

urnai/agents/rewards/vizdoom.py

@@ -41,12 +41,12 @@ def __init__(self, method):
     def get_reward(self, obs, reward, done):
         r = 0
 
-        if method == VizDoomHealthGatheringReward.METHOD_CUMULATIVE:
+        if self.method == VizDoomHealthGatheringReward.METHOD_CUMULATIVE:
             r += obs.game_variables[VizDoomHealthGatheringReward.HEALTH]
-        elif method == VizDoomHealthGatheringReward.METHOD_DIFFERENCE:
+        elif self.method == VizDoomHealthGatheringReward.METHOD_DIFFERENCE:
             r += obs.game_variables[VizDoomHealthGatheringReward.HEALTH] - self.prev_health 
             self.prev_health = obs.game_variables[VizDoomHealthGatheringReward.HEALTH]
-        elif method == VizDoomHealthGatheringReward.METHOD_POSITIVE_ONLY:
+        elif self.method == VizDoomHealthGatheringReward.METHOD_POSITIVE_ONLY:
             r += obs.game_variables[VizDoomHealthGatheringReward.HEALTH] - self.prev_health 
             self.prev_health = obs.game_variables[VizDoomHealthGatheringReward.HEALTH]