dodox/Columbus
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Small bug fix regarding aux_penalty

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-10-26 17:40:24 +02:00
parent e5ac7e0b05
commit 0fe5c35dda
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
columbus/env.py
View File

@ -242,7 +242,7 @@ class ColumbusEnv(gym.Env):
or self.agent.pos[1] < 0.001 or self.agent.pos[1] > 0.999:
reward -= self.void_damage/self.fps
self.score += reward # aux_reward does not count towards the score
if self.aux_reward_max:
if self.aux_reward_max or self.aux_penalty_max:
reward += self._get_aux_reward()
self._steps += 1
done = (self.die_on_zero and self.score <= 0) or (self.return_on_score != -