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

Fix converting list of LazyFrames to ndarray (#907)

Browse Source
This commit is contained in:
Anton Grigoryev
2019-06-01 02:49:46 +03:00
committed by pzhokhov
parent 1c872ca8fd
commit 7c520852d9
1 changed files with 7 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
baselines/common/atari_wrappers.py
View File

@@ -254,6 +254,13 @@ class LazyFrames(object):
return len(self._force()) return len(self._force())
def __getitem__(self, i): def __getitem__(self, i):
return self._force()[i]
def count(self):
frames = self._force()
return frames.shape[frames.ndim - 1]
def frame(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):