code: def dtype(dtype_name) -> torch.dtype:
      return TEST_DTYPES[dtype_name]
  
imports: import sys
  from itertools import product
  from typing import Dict
  import numpy
  import pytest
  import torch
  import kornia
  from kornia.utils._compat import torch_version
label_desc: Use image node_id 4 for calling a global function with example usage: dtype(dtype_name) and returns: TEST_DTYPES