ray/rllib/agents/impala/__init__.py

6 lines
221 B
Python

from ray.rllib.agents.impala.impala import ImpalaTrainer, DEFAULT_CONFIG
from ray.rllib.utils import renamed_agent
ImpalaAgent = renamed_agent(ImpalaTrainer)
__all__ = ["ImpalaAgent", "ImpalaTrainer", "DEFAULT_CONFIG"]