30 lines
657 B
Python
30 lines
657 B
Python
"""Env package."""
|
|
|
|
from tianshou.env.gym_wrappers import ContinuousToDiscrete, MultiDiscreteToDiscrete
|
|
from tianshou.env.venv_wrappers import VectorEnvNormObs, VectorEnvWrapper
|
|
from tianshou.env.venvs import (
|
|
BaseVectorEnv,
|
|
DummyVectorEnv,
|
|
RayVectorEnv,
|
|
ShmemVectorEnv,
|
|
SubprocVectorEnv,
|
|
)
|
|
|
|
try:
|
|
from tianshou.env.pettingzoo_env import PettingZooEnv
|
|
except ImportError:
|
|
pass
|
|
|
|
__all__ = [
|
|
"BaseVectorEnv",
|
|
"DummyVectorEnv",
|
|
"SubprocVectorEnv",
|
|
"ShmemVectorEnv",
|
|
"RayVectorEnv",
|
|
"VectorEnvWrapper",
|
|
"VectorEnvNormObs",
|
|
"PettingZooEnv",
|
|
"ContinuousToDiscrete",
|
|
"MultiDiscreteToDiscrete",
|
|
]
|