ray.rllib.core.rl_module.rl_module.RLModule.input_specs_inference#

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

Returns the input specs of the forward_inference method.