def get_name_to_features(
      config: ml_collections.ConfigDict  # pylint: disable=unused-argument
  ) -> Dict[str, Any]:
    """Return feature dict for decoding purposes.

    Models are responsible for preprocessing and therefore should
    already know what raw features are coming from the pipeline.

    Args:
      config: experiment config.

    Returns:
      Dict mapping feature names to feature types.
    """
    raise NotImplementedError
Return feature dict for decoding purposes.

    Models are responsible for preprocessing and therefore should
    already know what raw features are coming from the pipeline.

    Args:
      config: experiment config.

    Returns:
      Dict mapping feature names to feature types.
    
language/mentionmemory/tasks/base_task.py