from ray.rllib.agents.trainer import Trainer from ray.rllib.utils import renamed_agent Agent = renamed_agent(Trainer)