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

Allow manual reseting of path via arg to reset()

Browse Source
This commit is contained in:
Dominik Moritz Roth 2022-10-16 17:51:05 +02:00
parent 66c9509c27
commit 70536b79e7
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
columbus/env.py
View File

@ -266,7 +266,7 @@ class ColumbusEnv(gym.Env):
self.agent.pos = self.start_pos self.agent.pos = self.start_pos
# Expand this function # Expand this function
def reset(self): def reset(self, force_reset_path=False):
pygame.init() pygame.init()
self._init = True self._init = True
self._steps = 0 self._steps = 0
@ -286,7 +286,7 @@ 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()
if self.clear_path_on_reset: if self.clear_path_on_reset or force_reset_path:
self._reset_paths() self._reset_paths()
return self.observable.get_observation() return self.observable.get_observation()