code: def get_test_devices() -> Dict[str, torch.device]:
      """Create a dictionary with the devices to test the source code. CUDA devices will be test only in case the
      current hardware supports it.
  
      Return:
          dict(str, torch.device): list with devices names.
      """
      devices: Dict[str, torch.device] = {}
      devices["cpu"] = torch.device("cpu")
      if torch.cuda.is_available():
          devices["cuda"] = torch.device("cuda:0")
      if kornia.xla_is_available():
          import torch_xla.core.xla_model as xm
  
          devices["tpu"] = xm.xla_device()
      if hasattr(torch.backends, "mps"):
          if torch.backends.mps.is_available():
              devices["mps"] = torch.device("mps")
      return devices
  
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 1 for calling a global function with example usage: get_test_devices() and returns: devices