def _get_partition_for_machine()

in cli/healthscan.py [0:0]

• 12 lines of code
• 1 McCabe index (conditional complexity)

def _get_partition_for_machine(machine_type: str) -> str | None:
  """Returns the partition for the given orchestrator."""
  match machine_type:
    case 'a3-highgpu-8g':
      return 'a3'
    case 'a3-megagpu-8g':
      return 'a3mega'
    case 'a3-ultragpu-8g':
      return 'a3ultra'
    case 'a4-highgpu-8g':
      return 'a4'
    case _:
      raise ValueError(f'Unsupported machine type: {machine_type}')