ray.rllib.core.rl_module.rl_module.RLModule.output_specs_exploration#

RLModule.output_specs_exploration() List[str | Tuple[str, ...]] | NestedDict[Constraint | None][source]#

Returns the output specs of the forward_exploration() method.

Override this method to customize the output specs of the exploration call. The default implementation requires the forward_exploration() method to return a dict that has action_dist key and its value is an instance of Distribution.