code: def get_module(use_v2):
      # We need a protected import to avoid the V2 warning in case just V1 is used
      if use_v2:
          import torchvision.transforms.v2
  
          return torchvision.transforms.v2
      else:
          import torchvision.transforms
  
          return torchvision.transforms
  
file_path: vision/references/classification/presets.py
imports: import torch
  from torchvision.transforms.functional import InterpolationMode
label_desc: Use image node_id 1 for calling a global function with example usage: get_module(use_v2) and returns: torchvision, torchvision