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

    def preprocess_fn(sample: Dict[str, tf.Tensor]) -> Dict[str, tf.Tensor]:
      return sample

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