def accuracy_nr_fn(step):
  """Returns if predictions are correct."""
  return [
      pred == groundtruth for pred, groundtruth in zip(
          step.state.selected_applicants, step.state.true_eligible)
  ]