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

flake8 fix

Browse Source
This commit is contained in:
Peter Zhokhov
2019-01-24 14:33:50 -08:00
parent bb05b9ee88
commit 547764efc9
1 changed files with 0 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
baselines/common/vec_env/subproc_vec_env.py
View File

@@ -111,7 +111,6 @@ def _flatten_obs(obs):
assert len(obs) > 0 assert len(obs) > 0
if isinstance(obs[0], dict): if isinstance(obs[0], dict):
import collections
keys = obs[0].keys() keys = obs[0].keys()
return {k: np.stack([o[k] for o in obs]) for k in keys} return {k: np.stack([o[k] for o in obs]) for k in keys}
else: else: