code: def write_augmented_dataset(
      input_conllu, output_conllu, augment_function
  ):
      # set the seed for each data file so that the results are the same
      # regardless of how many treebanks are processed at once
      random.seed(1234)
  
      # read and shuffle conllu data
      sents = read_sentences_from_conllu(input_conllu)
  
      # the actual meat of the function - produce new sentences
      new_sents = augment_function(sents)
  
      write_sentences_to_conllu(output_conllu, new_sents)
  
imports: import argparse
  import glob
  import os
  import random
  import re
  import tempfile
  from collections import Counter
  from stanza.models.common.constant import treebank_to_short_name
  import stanza.utils.datasets.common
  from stanza.utils.datasets.common import read_sentences_from_conllu, write_sentences_to_conllu, INT_RE, MWT_RE, MWT_OR_COPY_RE
  import stanza.utils.datasets.tokenization.convert_ml_cochin
  import stanza.utils.datasets.tokenization.convert_my_alt
  import stanza.utils.datasets.tokenization.convert_vi_vlsp
  import stanza.utils.datasets.tokenization.convert_th_best
  import stanza.utils.datasets.tokenization.convert_th_lst20
  import stanza.utils.datasets.tokenization.convert_th_orchid
label_desc: Use image node_id 19 for calling a global function with example usage: write_augmented_dataset(input_conllu, output_conllu, augment_function) without return types