def get_agent_class(alg: str, return_config=False) -> type: try: return _get_trainer_class(alg, return_config=return_config) except ImportError: from ray.rllib.agents.mock import _trainer_import_failed class_ = _trainer_import_failed(traceback.format_exc()) config = class_.get_default_config() if return_config: return class_, config return class_
def get_trainer_class(alg: str, return_config=False) -> type: """Returns the class of a known Trainer given its name.""" try: return _get_trainer_class(alg, return_config=return_config) except ImportError: from ray.rllib.agents.mock import _trainer_import_failed class_ = _trainer_import_failed(traceback.format_exc()) config = class_._default_config if return_config: return class_, config return class_