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

MultiAgentRLModule.get_state(module_ids: Set[str] | None = None) → Mapping[str, Any][source]#

Returns the state of the multi-agent module.

This method returns the state of each module specified by module_ids. If module_ids is None, the state of all modules is returned.

Parameters:: module_ids – The module IDs to get the state of. If None, the state of all modules is returned.
Returns:: A nested state dict with the first layer being the module ID and the second is the state of the module. The returned dict values are framework-specific tensors.