ray.rllib.core.rl_module.marl_module.MultiAgentRLModule.forward_inference

MultiAgentRLModule.forward_inference(batch: SampleBatch | MultiAgentBatch, **kwargs) → Mapping[str, Any]

Forward-pass during evaluation, called from the sampler.

This method should not be overriden to implement a custom forward inference method. Instead, override the _forward_inference method.

Parameters:

• batch – The input batch. This input batch should comply with input_specs_inference().

• **kwargs – Additional keyword arguments.

Returns:

The output of the forward pass. This output should comply with the ouptut_specs_inference().