def METHOD_NAME(
    eval_config: config_pb2.EvalConfig,
    eval_shared_model: types.MaybeMultipleEvalSharedModels,
    output_batch_size: Optional[int] = None,
    output_keypath: Iterable[str] = (constants.PREDICTIONS_KEY,),