def is_valid_gpu_version()

in xlml/utils/xpk.py [0:0]

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

def is_valid_gpu_version(accelerator_type: str):
  if accelerator_type in [member.value for member in GpuVersion]:
    return True
  return False