code: def build_model(pretrain_file, *args):
      # By default, we turn off multistage, since that can turn off various other structures in the initial training
      args = [
          "--no_multistage",
          "--pattn_num_layers",
          "4",
          "--pattn_d_model",
          "256",
          "--hidden_size",
          "128",
          "--use_lattn",
      ] + list(args)
      trainer = build_trainer(pretrain_file, *args)
      return trainer.model
  
imports: import os
  import pytest
  import torch
  from stanza.models.common import pretrain
  from stanza.models.common.utils import set_random_seed
  from stanza.models.constituency import parse_transitions
  from stanza.tests import *
  from stanza.tests.constituency import test_parse_transitions
  from stanza.tests.constituency.test_trainer import build_trainer
label_desc: Use image node_id 2 for calling a global function with example usage: build_model(pretrain_file) and returns: trainer