code: def write_dataset(datasets, output_dir, dataset_name):
      for dataset, shard in zip(datasets, SHARDS):
          output_filename = os.path.join(
              output_dir, "%s_%s.mrg" % (dataset_name, shard)
          )
          print(
              "Writing {} trees to {}".format(
                  len(dataset), output_filename
              )
          )
          parse_tree.Tree.write_treebank(dataset, output_filename)
  
imports: import os
  from stanza.models.constituency import parse_tree
label_desc: Use image node_id 1 for calling a global function with example usage: write_dataset(datasets, output_dir, dataset_name) without return types