code: def prepare_tokenizer_treebank_labels(tokenizer_dir, short_name):
      """
      Given the txt and gold.conllu files, prepare mwt and label files for train/dev/test
      """
      for dataset in ("train", "dev", "test"):
          output_txt = f"{tokenizer_dir}/{short_name}.{dataset}.txt"
          output_conllu = (
              f"{tokenizer_dir}/{short_name}.{dataset}.gold.conllu"
          )
          try:
              prepare_tokenizer_dataset_labels(
                  output_txt,
                  output_conllu,
                  tokenizer_dir,
                  short_name,
                  dataset,
              )
          except (KeyboardInterrupt, SystemExit):
              raise
          except:
              print(
                  "Failed to convert %s to %s"
                  % (output_txt, output_conllu)
              )
              raise
  
imports: import argparse
  from enum import Enum
  import glob
  import logging
  import os
  import re
  import subprocess
  import sys
  from stanza.models.common.short_name_to_treebank import canonical_treebank_name
  import stanza.utils.datasets.prepare_tokenizer_data
  import stanza.utils.default_paths
label_desc: Use image node_id 5 for calling a global function with example usage: prepare_tokenizer_treebank_labels(tokenizer_dir, short_name) without return types