def get_distributed_eval_data(data_fn, strategy):
  """Gets a properly distributed evaluation data iterator."""
  _, eval_data = data_fn()
  dist_eval_data_iter = None
  if eval_data:
    dist_eval_data_iter = iter(
        strategy.distribute_datasets_from_function(lambda: eval_data))
  return dist_eval_data_iter