code: def test_lstm_tree_forward(pretrain_file):
      """
      Test the LSTM_TREE forward pass
      """
      model = build_model(
          pretrain_file,
          "--num_tree_lstm_layers",
          "1",
          "--constituency_composition",
          "tree_lstm",
      )
      run_forward_checks(model)
      model = build_model(
          pretrain_file,
          "--num_tree_lstm_layers",
          "2",
          "--constituency_composition",
          "tree_lstm",
      )
      run_forward_checks(model)
      model = build_model(
          pretrain_file,
          "--num_tree_lstm_layers",
          "3",
          "--constituency_composition",
          "tree_lstm",
      )
      run_forward_checks(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 35 for calling a global function with example usage: test_lstm_tree_forward(pretrain_file) without return types