add disable kernel Update in schema

Author: Himani Anil Deshpande

cli/src/pcluster/schemas/imagebuilder_schema.py

@@ -20,6 +20,7 @@
 from pcluster.config.imagebuilder_config import (
     Build,
     Component,
+    DisableKernelUpdate,
     DistributionConfiguration,
     Iam,
     Image,
@@ -170,6 +171,17 @@ def make_resource(self, data, **kwargs):
         return UpdateOsPackages(**data)
 
 
+class DisableKernelUpdateSchema(BaseSchema):
+    """Represent the schema of the ImageBuilder DisableKernelUpdate."""
+
+    enabled = fields.Bool()
+
+    @post_load
+    def make_resource(self, data, **kwargs):
+        """Generate resource."""
+        return DisableKernelUpdate(**data)
+
+
 class LustreClientSchema(BaseSchema):
     """Represent the schema of the ImageBuilder NvidiaSoftware."""
 
@@ -215,6 +227,7 @@ class BuildSchema(BaseSchema):
     security_group_ids = fields.List(fields.Str)
     subnet_id = fields.Str(validate=get_field_validator("subnet_id"))
     update_os_packages = fields.Nested(UpdateOsPackagesSchema)
+    disable_kernel_update = fields.Nested(DisableKernelUpdateSchema)
     imds = fields.Nested(ImdsSchema)
     installation = fields.Nested(InstallationSchema)
 

cli/tests/pcluster/schemas/test_imagebuilder_schema/test_imagebuilder_schema/imagebuilder_schema_all.yaml

@@ -37,6 +37,8 @@ Build:
   SubnetId: subnet-0d03dc52
   UpdateOsPackages:
     Enabled: true
+  DisableKernelUpdate:
+    Enabled: true
   Installation:
     NvidiaSoftware:
       Enabled: true
@@ -57,7 +59,6 @@ DevSettings:
       { "UserIds": ["123456789012", "345678901234"], "UserGroups": ["all"] }
   TerminateInstanceOnFailure: True
   DisableValidateAndTest: True
-  DisableKernelUpdate: True
 
 DeploymentSettings:
   {% if disable_sudo_access_for_default_user == "True" %}DisableSudoAccessForDefaultUser: True{% endif %}