more formatting

Author: Raymond Liu

src/sagemaker/inference_recommender/inference_recommender_mixin.py

@@ -149,7 +149,9 @@ def right_size(
         self._init_sagemaker_session_if_does_not_exist()
 
         self.temp_model_name = None
-        if isinstance(self, sagemaker.model.Model) and not isinstance(self, sagemaker.model.ModelPackage):
+        if isinstance(self, sagemaker.model.Model) and not isinstance(
+            self, sagemaker.model.ModelPackage
+        ):
 
             unique_tail = uuid.uuid4()
             self.temp_model_name = "SMPYTHONSDK-" + str(unique_tail)
@@ -160,10 +162,13 @@ def right_size(
                 container_defs=None,
                 primary_container=self.prepare_container_def(),
                 vpc_config=self.vpc_config,
-                enable_network_isolation=self.enable_network_isolation()
+                enable_network_isolation=self.enable_network_isolation(),
+            )
+            print(
+                f"Creating temporary model with name: {self.temp_model_name}"
+                " for Inference Recommender.",
+                flush=True,
             )
-            print(f"Creating temporary model with name: {self.temp_model_name}"
-                   " for Inference Recommender.", flush=True)
             self.sagemaker_session.create_model(**create_model_args)
             print("Temporary model created. Start to run Inference Recommender...", flush=True)
 
@@ -195,8 +200,11 @@ def right_size(
         )
 
         if self.temp_model_name is not None:
-            print(f"Deleting temporary model with name: {self.temp_model_name} "
-                   "for Inference Recommender.", flush=True)
+            print(
+                f"Deleting temporary model with name: {self.temp_model_name} "
+                "for Inference Recommender.",
+                flush=True,
+            )
             self.sagemaker_session.delete_model(self.temp_model_name)
             self.temp_model_name = None
             print("Delete complete.")

src/sagemaker/session.py

@@ -4974,12 +4974,15 @@ def create_inference_recommendations_job(
         """
 
         if model_name is None and model_package_version_arn is None:
-            raise ValueError("Missing model_name and model_package_version_arn,"
-                             " please provide one of them.")
+            raise ValueError(
+                "Missing model_name and model_package_version_arn," " please provide one of them."
+            )
 
         if model_name is not None and model_package_version_arn is not None:
-            raise ValueError("Please provide either model_name or model_package_version_arn"
-                             " should be provided, not both.")
+            raise ValueError(
+                "Please provide either model_name or model_package_version_arn"
+                " should be provided, not both."
+            )
 
         if not job_name:
             unique_tail = uuid.uuid4()

tests/integ/test_inference_recommender.py

@@ -184,9 +184,7 @@ def default_right_sized_unregistered_model(sagemaker_session, cpu_instance_type)
                 ir_job_name,
             )
         except Exception:
-            sagemaker_session.delete_model(
-                ModelName=sklearn_model.temp_model_name
-            )
+            sagemaker_session.delete_model(ModelName=sklearn_model.temp_model_name)
 
 
 @pytest.fixture(scope="module")
@@ -239,9 +237,7 @@ def advanced_right_sized_unregistered_model(sagemaker_session, cpu_instance_type
             )
 
         except Exception:
-            sagemaker_session.delete_model(
-                ModelName=sklearn_model.temp_model_name
-            )
+            sagemaker_session.delete_model(ModelName=sklearn_model.temp_model_name)
 
 
 @pytest.mark.slow_test

tests/unit/sagemaker/inference_recommender/test_inference_recommender_mixin.py

@@ -193,7 +193,7 @@ def test_right_size_default_with_model_name_successful(sagemaker_session, model)
         container_defs=None,
         primary_container={},
         vpc_config=None,
-        enable_network_isolation=False
+        enable_network_isolation=False,
     )
 
     # assert that the create api has been called with default parameters with model name

tests/unit/test_session.py

@@ -3284,7 +3284,9 @@ def create_inference_recommendations_job_advanced_happy_response():
 
 
 def create_inference_recommendations_job_advanced_model_name_happy_response():
-    base_advanced_job_response = create_inference_recommendations_job_default_model_name_happy_response()
+    base_advanced_job_response = (
+        create_inference_recommendations_job_default_model_name_happy_response()
+    )
 
     base_advanced_job_response["JobName"] = IR_JOB_NAME
     base_advanced_job_response["JobType"] = IR_ADVANCED_JOB
@@ -3394,7 +3396,7 @@ def test_create_inference_recommendations_job_advanced_model_name_happy(sagemake
 def test_create_inference_recommendations_job_missing_model_name_and_pkg(sagemaker_session):
     with pytest.raises(
         ValueError,
-        match="Missing model_name and model_package_version_arn, please provide one of them."
+        match="Missing model_name and model_package_version_arn, please provide one of them.",
     ):
         sagemaker_session.create_inference_recommendations_job(
             role=IR_ROLE_ARN,
@@ -3413,7 +3415,7 @@ def test_create_inference_recommendations_job_missing_model_name_and_pkg(sagemak
 def test_create_inference_recommendations_job_provided_model_name_and_pkg(sagemaker_session):
     with pytest.raises(
         ValueError,
-        match="Please provide either model_name or model_package_version_arn should be provided, not both."
+        match="Please provide either model_name or model_package_version_arn should be provided, not both.",
     ):
         sagemaker_session.create_inference_recommendations_job(
             role=IR_ROLE_ARN,