code: def device(device_name) -> torch.device:
      return TEST_DEVICES[device_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 3 for calling a global function with example usage: device(device_name) and returns: TEST_DEVICES