ray.rllib.policy.eager_tf_policy_v2.EagerTFPolicyV2.make_rl_module

EagerTFPolicyV2.make_rl_module() → RLModule

Returns the RL Module (only for when RLModule API is enabled.)

If RLModule API is enabled (self.config.rl_module(_enable_rl_module_api=True), this method should be implemented and should return the RLModule instance to use for this Policy. Otherwise, RLlib will error out.