code: def test_jit(self, device, dtype):
      points = torch.rand(1, 1, 2, device=device, dtype=dtype)
      K = torch.rand(1, 3, 3, device=device, dtype=dtype)
      new_K = torch.rand(1, 3, 3, device=device, dtype=dtype)
      distCoeff = torch.rand(1, 4, device=device, dtype=dtype)
      inputs = (points, K, distCoeff, new_K)
  
      op = distort_points
      op_jit = torch.jit.script(op)
      assert_close(op(*inputs), op_jit(*inputs))
  
file_path: kornia/test/geometry/calibration/test_distort.py
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  from kornia.geometry.calibration.distort import distort_points
  from kornia.testing import assert_close
label_desc: Use image node_id 5 for calling the TestDistortPoints obj's underlying member method code with example usage: obj.test_jit(device, dtype) without return types