code: def test_classy_model_wrapper_torch_scriptable(self):
      input = torch.ones((2, 2))
  
      for wrapper_cls, expected_output in [
          (None, input + 1),
          # this isn't supported yet
          # (TestSimpleClassyModelWrapper, (input + 1) * 2),
      ]:
          MyTestModel2.wrapper_cls = wrapper_cls
          model = MyTestModel2()
          scripted_model = torch.jit.script(model)
          self.assertTrue(torch.allclose(expected_output, model(input)))
          self.assertTrue(
              torch.allclose(expected_output, scripted_model(input))
          )
  
imports: import copy
  import os
  import shutil
  import tempfile
  import unittest
  import torch
  import torch.nn
  from classy_vision.generic.util import load_checkpoint
  from classy_vision.heads import FullyConnectedHead, IdentityHead
  from classy_vision.hooks import CheckpointHook
  from classy_vision.models import build_model, ClassyModel, ClassyModelWrapper, register_model
  from classy_vision.models.classy_model import _ClassyModelAdapter
  from classy_vision.tasks import build_task
  from classy_vision.trainer import LocalTrainer
  from test.generic.config_utils import get_fast_test_task_config, get_test_task_config
  from torchvision import models
label_desc: Use image node_id 7 for calling the TestClassyModel obj's underlying member method code with example usage: obj.test_classy_model_wrapper_torch_scriptable() without return types