a9f9940d17
- Refacor code to remove duplicate code - Enable async simulation for all vector envs - Remove `collector.close` and rename `VectorEnv` to `DummyVectorEnv` The abstraction of vector env changed. Prior to this pr, each vector env is almost independent. After this pr, each env is wrapped into a worker, and vector envs differ with their worker type. In fact, users can just use `BaseVectorEnv` with different workers, I keep `SubprocVectorEnv`, `ShmemVectorEnv` for backward compatibility. Co-authored-by: n+e <463003665@qq.com> Co-authored-by: magicly <magicly007@gmail.com>
12 lines
304 B
Python
12 lines
304 B
Python
from tianshou.env.worker.base import EnvWorker
|
|
from tianshou.env.worker.dummy import DummyEnvWorker
|
|
from tianshou.env.worker.subproc import SubprocEnvWorker
|
|
from tianshou.env.worker.ray import RayEnvWorker
|
|
|
|
__all__ = [
|
|
'EnvWorker',
|
|
'DummyEnvWorker',
|
|
'SubprocEnvWorker',
|
|
'RayEnvWorker',
|
|
]
|