code: def test_functional_call_with_jit(self, functional_call):
      module = MockModule()
      jit_module = torch.jit.script(module)
      with self.assertRaisesRegex(
          RuntimeError, r"used with Jitted modules"
      ):
          self._run_call_with_mock_module(jit_module, functional_call)
      x = torch.rand((1, 1))
      traced_module = torch.jit.trace(module, x)
      with self.assertRaisesRegex(
          RuntimeError, r"used with Jitted modules"
      ):
          self._run_call_with_mock_module(
              traced_module, functional_call
          )
  
imports: import contextlib
  import os
  import re
  import subprocess
  import sys
  import unittest
  import torch
  import torch.nn.utils.stateless
  from torch.testing._internal.common_cuda import TEST_MULTIGPU
  from torch.testing._internal.common_utils import run_tests, TestCase, parametrize, instantiate_parametrized_tests, subtest
label_desc: Use image node_id 4 for calling the TestStatelessFunctionalAPI obj's underlying member method code with example usage: obj.test_functional_call_with_jit(functional_call) without return types