code: def test_script(self, input_param, input_shape, _):
      net = UNETR(**(input_param))
      net.eval()
      with torch.no_grad():
          torch.jit.script(net)
  
      test_data = torch.randn(input_shape)
      test_script_save(net, test_data)
  
file_path: MONAI/tests/test_unetr.py
imports: from __future__ import annotations
  import unittest
  import torch
  from parameterized import parameterized
  from monai.networks import eval_mode
  from monai.networks.nets.unetr import UNETR
  from tests.utils import SkipIfBeforePyTorchVersion, skip_if_quick, test_script_save
label_desc: Use image node_id 3 for calling the TestUNETR obj's underlying member method code with example usage: obj.test_script(input_param, input_shape, _) without return types