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

Fixed bug with path drawing

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-09-23 22:22:51 +02:00
parent 29f96b1e8c
commit 678d6f620a
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
columbus/env.py
View File

@ -277,8 +277,13 @@ class ColumbusEnv(gym.Env):
self.setup() self.setup()
self.entities.append(self.agent) # add it last, will be drawn on top self.entities.append(self.agent) # add it last, will be drawn on top
self.observable.reset() self.observable.reset()
self._reset_paths()
return self.observable.get_observation() return self.observable.get_observation()
def _reset_paths(self):
self.path_overlay = pygame.Surface(
(self.width, self.height), pygame.SRCALPHA, 32)
def _draw_entities(self): def _draw_entities(self):
for entity in self.entities: for entity in self.entities:
entity.draw() entity.draw()