gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix dtype for wrapper observation spaces

Browse Source
This commit is contained in:
Karl Cobbe
2018-06-12 14:48:39 -07:00
parent a1d3c18ec0
commit 622915c473
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
baselines/common/atari_wrappers.py
View File

@@ -174,7 +174,7 @@ class FrameStack(gym.Wrapper):
class ScaledFloatFrame(gym.ObservationWrapper): class ScaledFloatFrame(gym.ObservationWrapper):
def __init__(self, env): def __init__(self, env):
gym.ObservationWrapper.__init__(self, env) gym.ObservationWrapper.__init__(self, env)
self.observation_space.dtype = np.float32 self.observation_space = gym.spaces.Box(low=0, high=1, shape=env.observation_space.shape, dtype=np.float32)
def observation(self, observation): def observation(self, observation):
# careful! This undoes the memory optimization, use # careful! This undoes the memory optimization, use