code: def test_jit(self, device, dtype):
      img1 = torch.rand(1, 3, 10, 10, device=device, dtype=dtype)
      img2 = torch.rand(1, 3, 10, 10, device=device, dtype=dtype)
  
      args = (img1, img2)
  
      op = kornia.losses.MS_SSIMLoss().to(device, dtype)
      op_script = torch.jit.script(op)
  
      assert_close(op(*args), op_script(*args))
  
imports: import pytest
  import torch
  from torch.autograd import gradcheck
  import kornia
  import kornia.testing
  from kornia.testing import BaseTester, assert_close
label_desc: Use image node_id 5 for calling the TestMS_SSIMLoss obj's underlying member method code with example usage: obj.test_jit(device, dtype) without return types