code: def write_multitag_dataset(
      datasets,
      output_dir,
      short_name,
      suffix="bio",
      shard_names=SHARDS,
      shards=SHARDS,
  ):
      for shard, dataset in zip(shard_names, datasets):
          output_filename = os.path.join(
              output_dir, "%s.%s.%s" % (short_name, shard, suffix)
          )
          write_sentences(output_filename, dataset)
  
      for shard, dataset in zip(shard_names, datasets):
          output_filename = os.path.join(
              output_dir, "%s.%s.json" % (short_name, shard)
          )
          write_multitag_json(output_filename, dataset)
  
imports: from collections import defaultdict
  import json
  import os
  import random
  from stanza.models.common.doc import Document
  import stanza.utils.datasets.ner.prepare_ner_file
label_desc: Use image node_id 6 for calling a global function with example usage: write_multitag_dataset(datasets, output_dir, short_name, suffix, shard_names, shards) without return types