ray.rllib.env.multi_agent_episode.MultiAgentEpisode.module_for

MultiAgentEpisode.module_for(agent_id: Hashable) → str | None

Returns the ModuleID for a given AgentID.

Forces the agent-to-module mapping to be performed (via self.agent_to_module_mapping_fn), if this has not been done yet. Note that all such mappings are stored in the self._agent_to_module_mapping property.

Parameters:

agent_id – The AgentID to get a mapped ModuleID for.

Returns:

The ModuleID mapped to from the given agent_id.