Smashed bugs regarding StateObservable giving wrong data when not

rendering
This commit is contained in:
2022-08-22 18:53:30 +02:00
parent bfbfe9bb43
commit f94eaa5dc0
3 changed files with 13 additions and 2 deletions
+3 -1
View File
@@ -192,6 +192,8 @@ class ColumbusEnv(gym.Env):
1 and self.score > self.return_on_score
info = {'score': self.score, 'reward': reward}
self._rendered = False
if done:
self.reset()
return observation, reward*self.reward_mult, done, info
def check_collisions_for(self, entity):
@@ -242,7 +244,7 @@ class ColumbusEnv(gym.Env):
self.agent = entities.Agent(self)
self.setup()
self.entities.append(self.agent) # add it last, will be drawn on top
self.observable._entities = None
self.observable.reset()
return self.observable.get_observation()
def _draw_entities(self):