Tianshou/tianshou/env/__init__.py
Mohammad Mahdi Rahimi c7e2e56fac
Pettingzoo support (#494)
Co-authored-by: Rodrigo de Lazcano <r.l.p.v96@gmail.com>
Co-authored-by: J K Terry <justinkterry@gmail.com>
2022-02-15 22:56:45 +08:00

20 lines
353 B
Python

"""Env package."""
from tianshou.env.pettingzoo_env import PettingZooEnv
from tianshou.env.venvs import (
BaseVectorEnv,
DummyVectorEnv,
RayVectorEnv,
ShmemVectorEnv,
SubprocVectorEnv,
)
__all__ = [
"BaseVectorEnv",
"DummyVectorEnv",
"SubprocVectorEnv",
"ShmemVectorEnv",
"RayVectorEnv",
"PettingZooEnv",
]