Tianshou/tianshou/env/__init__.py
Jiayi Weng c248b4f87e
fix conda support and keep API compatibility (#536)
* loose constrains

* fix nni issue (#478)

* fix coverage
2022-02-26 00:05:02 +08:00

24 lines
392 B
Python

"""Env package."""
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",
"PettingZooEnv",
]