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.


state_dict – The state dict to set.