Fix indexing LazyFrames (#875)

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

View File

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