ray.rllib.algorithms.algorithm.Algorithm.get_module

Algorithm.get_module(module_id: str = 'default_policy') → RLModule

Returns the (single-agent) RLModule with model_id (None if ID not found).

Parameters:

module_id – ID of the (single-agent) RLModule to return from the MARLModule used by the local EnvRunner.

Returns:

The SingleAgentRLModule sitting under the ModuleID key inside the local worker’s (EnvRunner’s) MARLModule.