2a7c151738
- add VectorEnvWrapper and VectorEnvNormObs - obs_rms store in policy save/load - align mujoco scripts with atari: obs_norm, envpool, wandb and README
27 lines
514 B
Python
27 lines
514 B
Python
"""Env package."""
|
|
|
|
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",
|
|
]
|