Bundle n_envs copies of make_env into a SyncVectorEnv.
(args, n_envs)
| 106 | |
| 107 | |
| 108 | def make_vec_env(args, n_envs): |
| 109 | """Bundle n_envs copies of make_env into a SyncVectorEnv.""" |
| 110 | return gym.vector.SyncVectorEnv([lambda: make_env(args) for _ in range(n_envs)]) |
| 111 | |
| 112 | |
| 113 | def pick_device(arg="auto"): |