def _select_exemplars(self, example: RawExample) -> Iterator[Sequence[str]]:
    """Yields a single list containing the top `max_exemplars` exemplars."""
    exemplar_hashed_ids = example["exemplars"]["hashed_ids"]
    for _ in range(self._converter_config.num_samples):
      yield exemplar_hashed_ids[:self._converter_config.max_exemplars]
Yields a single list containing the top `max_exemplars` exemplars.
language/casper/augment/casper_converters.py