Fixing black-check formats

Author: MadhubalasriB

src/sagemaker/session.py

@@ -2827,10 +2827,12 @@ def create_model_package_from_containers(
         )
         try:
             self.sagemaker_client.describe_model_package_group(
-                ModelPackageGroupName=request["ModelPackageGroupName"])
+                ModelPackageGroupName=request["ModelPackageGroupName"]
+            )
         except ClientError:
             self.sagemaker_client.create_model_package_group(
-                ModelPackageGroupName=request["ModelPackageGroupName"])
+                ModelPackageGroupName=request["ModelPackageGroupName"]
+            )
         return self.sagemaker_client.create_model_package(**request)
 
     def wait_for_model_package(self, model_package_name, poll=5):

src/sagemaker/workflow/_utils.py

@@ -439,7 +439,7 @@ def arguments(self) -> RequestType:
             description=self.description,
             tags=self.tags,
             container_def_list=self.container_def_list,
-            customer_metadata_properties=self.customer_metadata_properties
+            customer_metadata_properties=self.customer_metadata_properties,
         )
 
         request_dict = get_create_model_package_request(**model_package_args)

tests/integ/test_mxnet.py

@@ -253,7 +253,9 @@ def test_register_model_package_via_group(
             sagemaker_session=sagemaker_session,
             framework_version=mxnet_inference_latest_version,
         )
-        model_package_group_name = "register-model-package-{}".format(sagemaker_timestamp())
+        model_package_group_name = "register-model-package-{}".format(
+            sagemaker_timestamp()
+        )
         model_pkg = model.register(
             content_types=["application/json"],
             response_types=["application/json"],
@@ -266,12 +268,16 @@ def test_register_model_package_via_group(
         data = numpy.zeros(shape=(1, 1, 28, 28))
         result = predictor.predict(data)
         assert result is not None
-        model_packages = \
-            sagemaker_session.sagemaker_client.list_model_packages(ModelPackageGroupName=model_package_group_name)[
-                'ModelPackageSummaryList']
+        model_packages = sagemaker_session.sagemaker_client.list_model_packages(
+            ModelPackageGroupName=model_package_group_name
+        )["ModelPackageSummaryList"]
         for model_package in model_packages:
-            sagemaker_session.sagemaker_client.delete_model_package(ModelPackageName=model_package['ModelPackageArn'])
-        sagemaker_session.sagemaker_client.delete_model_package_group(ModelPackageGroupName=model_package_group_name)
+            sagemaker_session.sagemaker_client.delete_model_package(
+                ModelPackageName=model_package["ModelPackageArn"]
+            )
+        sagemaker_session.sagemaker_client.delete_model_package_group(
+            ModelPackageGroupName=model_package_group_name
+        )
 
 
 def test_register_model_package_versioned(

tests/unit/test_session.py

@@ -2416,7 +2416,7 @@ def test_create_model_package_from_containers_all_args(sagemaker_session):
         "CertifyForMarketplace": marketplace_cert,
         "ModelApprovalStatus": approval_status,
         "DriftCheckBaselines": drift_check_baselines,
-        "CustomerMetadataProperties": customer_metadata_properties
+        "CustomerMetadataProperties": customer_metadata_properties,
     }
     sagemaker_session.sagemaker_client.create_model_package.assert_called_with(**expected_args)