code: def test_basic_two_tag_training(pretrain_file, tmp_path):
      trainer = run_two_tag_training(pretrain_file, tmp_path)
      assert len(trainer.model.tag_clfs) == 2
      assert len(trainer.model.crits) == 2
      assert len(trainer.vocab["tag"].lens()) == 2
  
imports: import json
  import logging
  import os
  import warnings
  import pytest
  import torch
  from stanza.models import ner_tagger
  from stanza.models.ner.trainer import Trainer
  from stanza.tests import TEST_WORKING_DIR
  from stanza.utils.datasets.ner.prepare_ner_file import process_dataset
label_desc: Use image node_id 6 for calling a global function with example usage: test_basic_two_tag_training(pretrain_file, tmp_path) without return types