def METHOD_NAME(self, dataset: Dataset, is_train: bool) -> Dataset:
    # self.validate_schema(dataset, is_train)
    dataset = dataset.rename_columns(self._rename_fields)
    dataset = dataset.map(self._rename_examples,
                          load_from_cache_file=self._load_from_cache_file,
                          num_proc=self._num_workers
                          )
    dataset = dataset.map(self._convert_start_and_end_positions_from_bytes_to_chars,
                          load_from_cache_file=self._load_from_cache_file,
                          num_proc=self._num_workers
                          )
    dataset = super().METHOD_NAME(dataset, is_train)
    return dataset