RLModule.output_specs_inference() Union[List[Union[str, Tuple[str, ...]]], NestedDict[Optional[Constraint]]][source]#

Returns the output specs of the forward_inference method.

Override this method to customize the output specs of the inference call. The default implementation requires the forward_inference to reutn a dict that has action_dist key and its value is an instance of Distribution. This assumption must always hold.