code: def _get_unique_gpu_bases(images: List[Config]) -> Set[GpuBase]:
      gpu_base_images = set()
  
      for x in images:
          if x.get("gpu") is None:
              continue
          gpu_base_images.add(
              GpuBase(base_image=x["base_image"], gpu=x["gpu"])
          )
  
      return gpu_base_images
  
imports: from absl import flags, app, logging
  from copy import copy
  from typing import Dict, List, Any, NamedTuple, Optional, Set
  from enum import Enum
  import json
  import pprint
label_desc: Use image node_id 2 for calling a global function with example usage: _get_unique_gpu_bases(images) and returns: gpu_base_images