def _filter_treatment(dataset, seed):
  """Mapping function.

  Constructs bool variable (treated = True, control = False)
  Args:
    dataset: tf.data.Dataset
    seed: int
  Returns:
    dataset: tf.data.Dataset
  """
  t = False
  if dataset[f'image/sim_{seed}_pi/value'] == 1:
    t = True
  dataset['t'] = t
  return dataset