ray.rllib.policy.sample_batch.SampleBatch.as_multi_agent

SampleBatch.as_multi_agent(module_id: str | None = None) → MultiAgentBatch

Returns the respective MultiAgentBatch

Note, if module_id is not provided uses DEFAULT_POLICY`_ID.

Args;

module_id: An optional module ID. If None the DEFAULT_POLICY_ID

is used.

Returns:

The MultiAgentBatch (using DEFAULT_POLICY_ID) corresponding to this SampleBatch.