def load_splits():
  """Reads a JSON file containing split IDs.

  Returns:
    A dictionary where keys are a split name (e.g. `train` or `test`) and values
    are a list of integer example IDs.
  """
  with gfile.GFile(FLAGS.split, "r") as reader:
    text = reader.read()
  splits = json.loads(text)
  return splits
Reads a JSON file containing split IDs.

  Returns:
    A dictionary where keys are a split name (e.g. `train` or `test`) and values
    are a list of integer example IDs.
  
language/compgen/nqg/tasks/split_dataset.py