Fix indexing LazyFrames (#875)
Indexing LazyFrames with index i should return the single channel frame
This commit is contained in:
@@ -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)
|
||||
|
Reference in New Issue
Block a user