def _get_split(self, file_path):
    """Get the data split based on the filename suffix."""
    if file_path.endswith('train.c2s'):
      return constants.TRAIN_SPLIT_NAME
    elif file_path.endswith('val.c2s'):
      return constants.VALIDATION_SPLIT_NAME
    else:
      return constants.TEST_SPLIT_NAME