flake8 fix

This commit is contained in:
Peter Zhokhov
2019-01-24 14:33:50 -08:00
parent bb05b9ee88
commit 547764efc9

View File

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