Raises:
      DecodeError: If the sample is not RepresentativeDataSample.
    """
    self.dataset_file_map = dataset_file_map

  def _load_tf_record(self, tf_record_path: str) -> RepresentativeDataset:
    """Loads TFRecord containing samples of type`RepresentativeDataSample`."""
    samples = []
    with context.eager_mode():