code: def test_jit(self, batch_size, device, dtype):
      src, dst = _sample_points(batch_size, device)
      op = kornia.geometry.transform.get_tps_transform
      op_jit = torch.jit.script(op)
      op_output = op(src, dst)
      jit_output = op_jit(src, dst)
      assert_close(op_output[0], jit_output[0])
      assert_close(op_output[1], jit_output[1])
  
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia
  from kornia.testing import assert_close
  from kornia.utils._compat import torch_version_ge
label_desc: Use image node_id 6 for calling the TestTransformParameters obj's underlying member method code with example usage: obj.test_jit(batch_size, device, dtype) without return types