code: def get_dataset_split(tmp_dir, split, use_control_set):
    
    if not use_control_set:
      dataset_split = {
          problem.DatasetSplit.TRAIN: [
              f for f in tf.gfile.Glob(
                  os.path.join(tmp_dir, "train-novels/*/*.txt"))
          ],
          problem.DatasetSplit.EVAL: [
              os.path.join(tmp_dir, "lambada_development_plain_text.txt")
          ],
          problem.DatasetSplit.TEST: [
              os.path.join(tmp_dir, "lambada_test_plain_text.txt")
          ]
      }
  
    else:
      dataset_split = {
          problem.DatasetSplit.TRAIN: [
              f for f in tf.gfile.Glob(
                  os.path.join(tmp_dir, "train-novels/*/*.txt"))
          ],
          problem.DatasetSplit.EVAL: [
              os.path.join(tmp_dir, "lambada_control_test_data_plain_text.txt")
          ],
      }
  
    return dataset_split[split]
docs: Gives the file paths with regards to the given split.
  
    Args:
      tmp_dir: temp directory
      split: dataset split
      use_control_set: uses control dataset if true.
  
    Returns:
      list of file paths.