code: def test_script(self):
      # test 2D network
      net = AHNet(spatial_dims=2, out_channels=2)
      test_data = torch.randn(1, 1, 128, 64)
      test_script_save(net, test_data)
      # test 3D network
      net = AHNet(
          spatial_dims=3,
          out_channels=2,
          psp_block_num=0,
          upsample_mode="nearest",
      )
      test_data = torch.randn(1, 1, 32, 32, 64)
      test_script_save(net, test_data)
  
imports: from __future__ import annotations
  import unittest
  import torch
  from parameterized import parameterized
  from monai.networks import eval_mode
  from monai.networks.blocks import FCN, MCFCN
  from monai.networks.nets import AHNet
  from tests.utils import skip_if_quick, test_pretrained_networks, test_script_save
label_desc: Use image node_id 3 for calling the TestAHNET obj's underlying member method code with example usage: obj.test_script() without return types