MultiAgentRLModule.get_state(module_ids: Set[str] | None = None, inference_only: bool = False) 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.

  • module_ids – The module IDs to get the state of. If None, the state of all modules is returned.

  • inference_only – If True, only a subset of parameters that are needed for inference are returned. This subset is defined in the module.


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.