Files
ray/rllib/env/__init__.py
T
Eric Liang 1e4a1360fd [rllib] Add type annotations to Trainer class (#8642)
* type trainer

* type it

* fxi
2020-06-03 12:47:35 -07:00

27 lines
765 B
Python

from typing import Any
from ray.rllib.env.base_env import BaseEnv
from ray.rllib.env.dm_env_wrapper import DMEnv
from ray.rllib.env.multi_agent_env import MultiAgentEnv
from ray.rllib.env.external_env import ExternalEnv
from ray.rllib.env.external_multi_agent_env import ExternalMultiAgentEnv
from ray.rllib.env.vector_env import VectorEnv
from ray.rllib.env.env_context import EnvContext
from ray.rllib.env.policy_client import PolicyClient
from ray.rllib.env.policy_server_input import PolicyServerInput
# Represents one of the env types in this package.
EnvType = Any
__all__ = [
"BaseEnv",
"MultiAgentEnv",
"ExternalEnv",
"ExternalMultiAgentEnv",
"VectorEnv",
"EnvContext",
"DMEnv",
"PolicyClient",
"PolicyServerInput",
]