fix(codegen): restore codestyle-doc-tests compliance after botocore sync

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",

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__)
 
 

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__)
 
 """
 

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)}"
             )
 

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)

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": [