def _get_values_file()

in cli/gke_check.py [0:0]

• 13 lines of code
• 2 McCabe index (conditional complexity)

  def _get_values_file(self) -> str:
    """Get the values file for the check."""
    base_path = 'deploy/helm/health_runner/'
    match self.machine_type:
      case 'a3-highgpu-8g':
        return base_path + 'a3high.yaml'
      case 'a3-megagpu-8g':
        # Use the default values for A3 Mega
        return base_path + 'values.yaml'
      case 'a3-ultragpu-8g':
        return base_path + 'a3ultra.yaml'
      case 'a4-highgpu-8g':
        return base_path + 'a4.yaml'
      case _:
        raise ValueError(f'Unsupported machine type: {self.machine_type}')