Move Image Features to AzureImageSchema

lisa/feature.py

@@ -89,7 +89,7 @@ def create_setting(
 
     @classmethod
     def create_image_requirement(
-        cls, *args: Any, **kwargs: Any
+        cls, image: schema.ImageSchema
     ) -> Optional[schema.FeatureSettings]:
         """
         It's called in the platform to check if an image restricts the feature or not.

lisa/schema.py

@@ -870,10 +870,6 @@ class NodeSpace(search_space.RequirementMixin, TypedSchema, ExtendableSchemaMixi
         ),
     )
 
-    # Platform may add image features to
-    # keep track of platform requirements.
-    _image_features: Optional[Dict[str, Any]] = None
-
     def __post_init__(self, *args: Any, **kwargs: Any) -> None:
         # clarify types to avoid type errors in properties.
         self._features: Optional[search_space.SetSpace[FeatureSettings]]
@@ -1624,6 +1620,11 @@ class ImageSchema:
     pass
 
 
+class ArchitectureType(str, Enum):
+    x64 = "x64"
+    Arm64 = "Arm64"
+
+
 def load_by_type(schema_type: Type[T], raw_runbook: Any, many: bool = False) -> T:
     """
     Convert dict, list or base typed schema to specified typed schema.