flake8 fix
This commit is contained in:
@@ -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:
|
||||
|
Reference in New Issue
Block a user