From 547764efc9d77bc5c54b8ec872562bbbe79e545f Mon Sep 17 00:00:00 2001 From: Peter Zhokhov Date: Thu, 24 Jan 2019 14:33:50 -0800 Subject: [PATCH] flake8 fix --- baselines/common/vec_env/subproc_vec_env.py | 1 - 1 file changed, 1 deletion(-) diff --git a/baselines/common/vec_env/subproc_vec_env.py b/baselines/common/vec_env/subproc_vec_env.py index 134e371..e040e3d 100644 --- a/baselines/common/vec_env/subproc_vec_env.py +++ b/baselines/common/vec_env/subproc_vec_env.py @@ -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: