diff --git a/sagemaker-core/src/sagemaker/core/config_schema.py b/sagemaker-core/src/sagemaker/core/config_schema.py
index c064177f7b..de041068fb 100644
--- a/sagemaker-core/src/sagemaker/core/config_schema.py
+++ b/sagemaker-core/src/sagemaker/core/config_schema.py
@@ -1,3 +1,5 @@
+"""JSON schema definition for the SageMaker Python SDK configuration file."""
+
 SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA = {
     "$schema": "https://json-schema.org/draft/2020-12/schema",
     "type": "object",
diff --git a/sagemaker-core/src/sagemaker/core/resources.py b/sagemaker-core/src/sagemaker/core/resources.py
index c0780dc553..dc6bca5f63 100644
--- a/sagemaker-core/src/sagemaker/core/resources.py
+++ b/sagemaker-core/src/sagemaker/core/resources.py
@@ -46,7 +46,6 @@
 from sagemaker.core.serializers.base import BaseSerializer
 from sagemaker.core.deserializers.base import BaseDeserializer
 
-
 logger = get_textual_rich_logger(__name__)
 
 
diff --git a/sagemaker-core/src/sagemaker/core/tools/constants.py b/sagemaker-core/src/sagemaker/core/tools/constants.py
index 6d5580820c..fe07f0f0a4 100644
--- a/sagemaker-core/src/sagemaker/core/tools/constants.py
+++ b/sagemaker-core/src/sagemaker/core/tools/constants.py
@@ -78,8 +78,7 @@
 # language governing permissions and limitations under the License.
 """
 
-LOGGER_STRING = """
-logger = get_textual_rich_logger(__name__)
+LOGGER_STRING = """logger = get_textual_rich_logger(__name__)
 
 """
 
diff --git a/sagemaker-core/src/sagemaker/core/tools/resources_codegen.py b/sagemaker-core/src/sagemaker/core/tools/resources_codegen.py
index 22428f3006..f981001332 100644
--- a/sagemaker-core/src/sagemaker/core/tools/resources_codegen.py
+++ b/sagemaker-core/src/sagemaker/core/tools/resources_codegen.py
@@ -2115,6 +2115,7 @@ def generate_config_schema(self) -> str:
         with open(output, "w") as file:
             # Generate and write the license to the file
             file.write(
+                '"""JSON schema definition for the SageMaker Python SDK configuration file."""\n\n'
                 f"SAGEMAKER_PYTHON_SDK_CONFIG_SCHEMA = {json.dumps(combined_config_schema, indent=4)}"
             )
 
diff --git a/sagemaker-core/src/sagemaker/core/tools/shapes_extractor.py b/sagemaker-core/src/sagemaker/core/tools/shapes_extractor.py
index b92e50524c..fb35676810 100644
--- a/sagemaker-core/src/sagemaker/core/tools/shapes_extractor.py
+++ b/sagemaker-core/src/sagemaker/core/tools/shapes_extractor.py
@@ -43,6 +43,7 @@ def __init__(self, combined_shapes: Optional[dict] = None):
 
         self.shape_dag = self.get_shapes_dag()
         with open(SHAPE_DAG_FILE_PATH, "w") as f:
+            f.write('"""Generated shape dependency graph used for serialization."""\n\n')
             f.write("SHAPE_DAG=")
             f.write(textwrap.indent(pprint.pformat(self.shape_dag, width=1), "") + "\n")
         reformat_file_with_black(SHAPE_DAG_FILE_PATH)
diff --git a/sagemaker-core/src/sagemaker/core/utils/code_injection/shape_dag.py b/sagemaker-core/src/sagemaker/core/utils/code_injection/shape_dag.py
index ff5f80e6be..790d029333 100644
--- a/sagemaker-core/src/sagemaker/core/utils/code_injection/shape_dag.py
+++ b/sagemaker-core/src/sagemaker/core/utils/code_injection/shape_dag.py
@@ -1,3 +1,5 @@
+"""Generated shape dependency graph used for serialization."""
+
 SHAPE_DAG = {
     "AIAdapterModelPackageEntry": {
         "members": [