31 lines
687 B
Python
31 lines
687 B
Python
"""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",
|
|
]
|