gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00
Code Issues Projects Releases Wiki Activity

bug fix for dumping gifs from doom

Browse Source
This commit is contained in:
Itai Caspi
2017-12-20 13:09:41 +02:00
committed by Itai Caspi
parent 37e317682b
commit 9e59d1960e
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
environments/doom_environment_wrapper.py
View File

@@ -146,6 +146,11 @@ class DoomEnvironmentWrapper(EnvironmentWrapper):
def _preprocess_observation(self, observation): def _preprocess_observation(self, observation):
if observation is None: if observation is None:
return None return None
# for the last step we get no new observation, so we shouldn't preprocess it
if self.done:
return observation
# move the channel to the last axis # move the channel to the last axis
observation = np.transpose(observation, (1, 2, 0)) observation = np.transpose(observation, (1, 2, 0))
return observation return observation