code: def process_treebank(treebank, model_type, paths, args):
      short_name = treebank_to_short_name(treebank)
  
      mwt_dir = paths["MWT_DATA_DIR"]
      os.makedirs(mwt_dir, exist_ok=True)
  
      with tempfile.TemporaryDirectory() as tokenizer_dir:
          paths = dict(paths)
          paths["TOKENIZE_DATA_DIR"] = tokenizer_dir
  
          # first we process the tokenization data
          tokenizer_args = argparse.Namespace()
          tokenizer_args.augment = False
          tokenizer_args.prepare_labels = True
          prepare_tokenizer_treebank.process_treebank(
              treebank, model_type, paths, tokenizer_args
          )
  
          copy_conllu(tokenizer_dir, mwt_dir, short_name, "train", "in")
          copy_conllu(tokenizer_dir, mwt_dir, short_name, "dev", "gold")
          copy_conllu(
              tokenizer_dir, mwt_dir, short_name, "test", "gold"
          )
  
          for shard in ("train", "dev", "test"):
              source_filename = common.mwt_name(
                  tokenizer_dir, short_name, shard
              )
              dest_filename = common.mwt_name(
                  mwt_dir, short_name, shard
              )
              print(
                  "Copying from %s to %s"
                  % (source_filename, dest_filename)
              )
              shutil.copyfile(source_filename, dest_filename)
  
          contract_mwt(
              f"{mwt_dir}/{short_name}.dev.gold.conllu",
              f"{mwt_dir}/{short_name}.dev.in.conllu",
          )
          contract_mwt(
              f"{mwt_dir}/{short_name}.test.gold.conllu",
              f"{mwt_dir}/{short_name}.test.in.conllu",
          )
  
file_path: stanza/stanza/utils/datasets/prepare_mwt_treebank.py
imports: import argparse
  import os
  import shutil
  import tempfile
  from stanza.models.common.constant import treebank_to_short_name
  import stanza.utils.datasets.common
  import stanza.utils.datasets.prepare_tokenizer_treebank
  from stanza.utils.datasets.contract_mwt import contract_mwt
label_desc: Use image node_id 2 for calling a global function with example usage: process_treebank(treebank, model_type, paths, args) without return types