From 83eb759b9c7125f3ea99f445934c6db6c407571b Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Fri, 14 Oct 2022 15:23:40 +0200 Subject: [PATCH] Fix: Fatal Bug, max_steps always triggering --- columbus/env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/columbus/env.py b/columbus/env.py index dd780a5..f483761 100644 --- a/columbus/env.py +++ b/columbus/env.py @@ -219,7 +219,7 @@ class ColumbusEnv(gym.Env): reward += self._get_aux_reward() self._steps += 1 done = self._term_next or (self.die_on_zero and self.score <= 0) or (self.return_on_score != - - 1 and self.score > self.return_on_score) or self._steps >= self.max_steps + 1 and self.score > self.return_on_score) or self._steps == self.max_steps # make sure we register the current reward self._term_next = self.terminate_on_reward and gotRew info = {'score': self.score, 'reward': reward}