def get_accelerator_type()

in dags/map_reproducibility/utils/common_utils.py [0:0]

• 7 lines of code
• 4 McCabe index (conditional complexity)

def get_accelerator_type(hypercomputer: str):
  if hypercomputer == "a3ultra":
    return "h200"
  elif hypercomputer == "a3mega":
    return "h100"
  elif hypercomputer == "a4":
    return "b200"