ray.rllib.core.rl_module.marl_module.MultiAgentRLModule.set_state#

MultiAgentRLModule.set_state(state_dict: Mapping[str, Any]) None[source]#

Sets the state of the multi-agent module.

It is assumed that the state_dict is a mapping from module IDs to their corressponding state. This method sets the state of each module by calling their set_state method. If you want to set the state of some of the RLModules within this MultiAgentRLModule your state_dict can only include the state of those RLModules. Override this method to customize the state_dict for custom more advanced multi-agent use cases.

Parameters:

state_dict – The state dict to set.