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

Fix indexing LazyFrames (#875)

Browse Source 
Indexing LazyFrames with index i should return the single channel frame
This commit is contained in:
Taeyeong Jeong
2019-04-20 07:00:09 +09:00
committed by pzhokhov
parent fa37beb52e
commit 5d8041d18e
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

@@ -221,7 +221,7 @@ class LazyFrames(object):
return len(self._force()) return len(self._force())
def __getitem__(self, i): def __getitem__(self, i):
return self._force()[i] return self._force()[..., i]
def make_atari(env_id, max_episode_steps=None): def make_atari(env_id, max_episode_steps=None):
env = gym.make(env_id) env = gym.make(env_id)