# Replaces the dataset with a new dataset where tf.Tensors are replaced
      # by their evaluated values.
      ds = repr_ds_map[signature_def_key]
      repr_ds_map[signature_def_key] = rd.replace_tensors_by_numpy_ndarrays(
          ds, sess
      )


def _create_sample_validator(
    expected_input_keys: Collection[str],
) -> Callable[[rd.RepresentativeSample], rd.RepresentativeSample]: