def load_data(is_train, num_par=4):
  """Loads the tf.data.Dataset to run on."""
  if is_train:
    src = FLAGS.train_data_path
  else:
    src = FLAGS.dev_data_path

  if src is None:
    raise ValueError("Missing data path")

  if FLAGS.dataset == "boolq":
    return load_boolq_file(src, num_par)
  else:
    return load_nli_file(src, num_par)
Loads the tf.data.Dataset to run on.
language/boolq/run_recurrent_model_boolq.py