ray.rllib.algorithms.algorithm_config.AlgorithmConfig.rl_module#

AlgorithmConfig.rl_module(*, model_config_dict: ~typing.Dict[str, ~typing.Any] | None = <ray.rllib.utils.from_config._NotProvided object>, rl_module_spec: ~ray.rllib.core.rl_module.rl_module.SingleAgentRLModuleSpec | ~ray.rllib.core.rl_module.marl_module.MultiAgentRLModuleSpec | None = <ray.rllib.utils.from_config._NotProvided object>, _enable_rl_module_api=-1) → AlgorithmConfig[source]#

Sets the config’s RLModule settings.

Parameters:

  • model_config_dict – The default model config dictionary for RLModule`s. This will be used for any `RLModule if not otherwise specified in the rl_module_spec.

  • rl_module_spec – The RLModule spec to use for this config. It can be either a SingleAgentRLModuleSpec or a MultiAgentRLModuleSpec. If the observation_space, action_space, catalog_class, or the model config is not specified it will be inferred from the env and other parts of the algorithm config object.

Returns:

This updated AlgorithmConfig object.