From 61160c8595a060d8e79a28b4a0b69b15764f9cf4 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Fri, 14 Oct 2022 17:08:12 +0200
Subject: [PATCH] term_next hack actually not necessary

---
 columbus/env.py | 8 ++------
 1 file changed, 2 insertions(+), 6 deletions(-)

diff --git a/columbus/env.py b/columbus/env.py
index f483761..7f5bb92 100644
--- a/columbus/env.py
+++ b/columbus/env.py
@@ -107,7 +107,6 @@ class ColumbusEnv(gym.Env):
         self._seed(self.env_seed)
 
         self._init = False
-        self._term_next = False
 
     @property
     def observation_space(self):
@@ -218,10 +217,8 @@ class ColumbusEnv(gym.Env):
         if self.aux_reward_max:
             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
-        # make sure we register the current reward
-        self._term_next = self.terminate_on_reward and gotRew
+        done = (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) or (self.terminate_on_reward and gotRew)
         info = {'score': self.score, 'reward': reward}
         self._rendered = False
         if done:
@@ -268,7 +265,6 @@ class ColumbusEnv(gym.Env):
     def reset(self):
         pygame.init()
         self._init = True
-        self._term_next = False
         self._steps = 0
         self._seed(self.env_seed)
         self._rendered = False