def make_output_postprocess_fn(
      cls,
      config: ml_collections.ConfigDict  # pylint: disable=unused-argument
  ) -> Callable[[Dict[str, Any], Dict[str, Any]], Dict[str, Any]]:
    """Postprocess task samples (input and output). See BaseTask."""

    return MemoryRetrievalResultProcessor(config)
Postprocess task samples (input and output). See BaseTask.
language/mentionmemory/encoders/mention_memory_encoder.py