code formatting

Author: Himani Anil Deshpande

cli/src/pcluster/config/imagebuilder_config.py

@@ -148,8 +148,8 @@ class DisableKernelUpdate(Resource):
     """Represent the DisableKernelUpdate configuration for the ImageBuilder."""
 
     def __init__(
-            self,
-            enabled: bool = None,
+        self,
+        enabled: bool = None,
     ):
         super().__init__()
         self.enabled = Resource.init_param(enabled, default=True)
@@ -362,7 +362,9 @@ def _set_default(self, config: ImageBuilderConfig):
             dev_settings.slurm_patches_s3_archive if dev_settings and dev_settings.slurm_patches_s3_archive else ""
         )
         self.base_os = "{{ build.OperatingSystemName.outputs.stdout }}"
-        self.disable_kernel_update = {"enabled": "yes"} if config.build.disable_kernel_update.enabled else {"enabled": "no"}
+        self.disable_kernel_update = (
+            {"enabled": "yes"} if config.build.disable_kernel_update.enabled else {"enabled": "no"}
+        )
         for key, value in self.__dict__.items():
             if not key.startswith("_") and key not in self._cluster_attributes:
                 self._cluster_attributes.update({key: value})

cli/src/pcluster/templates/imagebuilder_stack.py

@@ -388,13 +388,21 @@ def _add_imagebuilder_distribution_configuration(self, ami_tags, build_tags, lam
     def _add_imagebuilder_user_data_override(self):
 
         additional_instance_configuration_property = None
-        if self.config.build and self.config.build.disable_kernel_update and self.config.build.disable_kernel_update.enabled:
-            imagebuilder_user_data = os.path.join(imagebuilder_utils.get_resources_directory(), "imagebuilder", "user_data.sh")
+        if (
+            self.config.build
+            and self.config.build.disable_kernel_update
+            and self.config.build.disable_kernel_update.enabled
+        ):
+            imagebuilder_user_data = os.path.join(
+                imagebuilder_utils.get_resources_directory(), "imagebuilder", "user_data.sh"
+            )
             with open(imagebuilder_user_data, "r", encoding="utf-8") as user_data_file:
                 user_data_content = user_data_file.read()
 
-            additional_instance_configuration_property=imagebuilder.CfnImageRecipe.AdditionalInstanceConfigurationProperty(
-                user_data_override=Fn.base64(user_data_content)
+            additional_instance_configuration_property = (
+                imagebuilder.CfnImageRecipe.AdditionalInstanceConfigurationProperty(
+                    user_data_override=Fn.base64(user_data_content)
+                )
             )
         return additional_instance_configuration_property
 
@@ -432,7 +440,6 @@ def _add_imagebuilder_image_recipe(self, build_tags, components, lambda_cleanup_
 
         return image_recipe_resource
 
-
     def _add_imagebuilder_components(self, build_tags, lambda_cleanup_policy_statements):
         imagebuilder_resources_dir = os.path.join(imagebuilder_utils.get_resources_directory(), "imagebuilder")
 

cli/tests/pcluster/config/dummy_imagebuilder_config.py

@@ -13,6 +13,7 @@
     AdditionalIamPolicy,
     Build,
     Component,
+    DisableKernelUpdate,
     DistributionConfiguration,
     Iam,
     Image,
@@ -23,7 +24,7 @@
     LustreClient,
     NvidiaSoftware,
     UpdateOsPackages,
-    Volume, DisableKernelUpdate,
+    Volume,
 )
 
 CLASS_DICT = {