File size: 687 Bytes
9b19c29 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
"""Env package."""
from tianshou.env.gym_wrappers import (
ContinuousToDiscrete,
MultiDiscreteToDiscrete,
TruncatedAsTerminated,
)
from tianshou.env.pettingzoo_env import PettingZooEnv
from tianshou.env.venv_wrappers import VectorEnvNormObs, VectorEnvWrapper
from tianshou.env.venvs import (
BaseVectorEnv,
DummyVectorEnv,
RayVectorEnv,
ShmemVectorEnv,
SubprocVectorEnv,
)
__all__ = [
"BaseVectorEnv",
"DummyVectorEnv",
"SubprocVectorEnv",
"ShmemVectorEnv",
"RayVectorEnv",
"VectorEnvWrapper",
"VectorEnvNormObs",
"PettingZooEnv",
"ContinuousToDiscrete",
"MultiDiscreteToDiscrete",
"TruncatedAsTerminated",
]
|