def make_preprocess_fn(
      config: ml_collections.ConfigDict
  ) -> Callable[[Dict[Text, tf.Tensor]], Dict[Text, tf.Tensor]]:
    """Produces function to preprocess samples. See BaseTask."""

    def preprocess_fn(example: Dict[Text, tf.Tensor]) -> Dict[Text, tf.Tensor]:
      return example

    return preprocess_fn
Produces function to preprocess samples. See BaseTask.
language/mentionmemory/tasks/text_classifier.py