code: def test_jit(self, device, dtype):
      input1 = torch.linspace(
          0, 255, 10, device=device, dtype=dtype
      ).unsqueeze(0)
      bins = torch.linspace(0, 255, 10, device=device, dtype=dtype)
      bandwidth = torch.tensor(2 * 0.4**2, device=device, dtype=dtype)
      inputs = (input1, bins, bandwidth)
  
      op = TestHistogram.fcn
      op_script = torch.jit.script(op)
  
      assert_close(op(*inputs), op_script(*inputs))
  
imports: import pytest
  import torch
  from packaging import version
  from torch.autograd import gradcheck
  import kornia
  import kornia.testing
  from kornia.testing import assert_close
label_desc: Use image node_id 4 for calling the TestHistogram obj's underlying member method code with example usage: obj.test_jit(device, dtype) without return types