Accept two different keys for Architecture Type

I have seen both tags for architecture type. Check both for a more robust solution.
microsoft · May 24, 2024 · bab6cab · bab6cab
1 parent 745d6d3
commit bab6cab
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/lisa/sut_orchestrator/azure/features.py b/lisa/sut_orchestrator/azure/features.py
@@ -3077,9 +3077,12 @@ def create_setting(
  cls, *args: Any, **kwargs: Any
  ) -> Optional[schema.FeatureSettings]:
  raw_capabilities: Any = kwargs.get("raw_capabilities")
- return ArchitectureSettings(
- arch=raw_capabilities.get("Cpuschema.ArchitectureType", "x64")
+ arch: schema.ArchitectureType = schema.ArchitectureType(
+ raw_capabilities.get("CpuArchitectureType")
+ or raw_capabilities.get("Cpuschema.ArchitectureType")
+ or "x64"
  )
+ return ArchitectureSettings(arch=arch)
 
  @classmethod
  def create_image_requirement(