style: apply black formatting to sagemaker-core and sagemaker-serve unit tests

---
X-AI-Prompt: Run formatters and fix issues
X-AI-Tool: Claude

Author: Joseph Zhang

sagemaker-core/tests/unit/local/test_entities.py

@@ -762,7 +762,9 @@ def test_wait_custom_timeout(self, mock_sleep, mock_get_host, mock_perform_reque
     @patch("sagemaker.core.local.entities._perform_request")
     @patch("sagemaker.core.local.entities.get_docker_host")
     @patch("time.sleep")
-    def test_wait_uses_default_timeout_constant(self, mock_sleep, mock_get_host, mock_perform_request):
+    def test_wait_uses_default_timeout_constant(
+        self, mock_sleep, mock_get_host, mock_perform_request
+    ):
         """Test that _wait_for_serving_container defaults to HEALTH_CHECK_TIMEOUT_LIMIT"""
         mock_get_host.return_value = "localhost"
         # Succeed on first attempt so we can verify it was called

sagemaker-serve/tests/unit/conftest.py

@@ -26,6 +26,7 @@
 the suite three times (py310/py311/py312) and these failures almost always hit
 only one of the three.
 """
+
 from __future__ import annotations
 
 from contextlib import ExitStack

sagemaker-serve/tests/unit/test_jumpstart_default_instance_type.py

@@ -35,9 +35,7 @@ def _load_fixture_spec_dict():
     # Exclude ml.m5.large (the buggy code's hardcoded fallback) from the
     # supported list so membership assertions discriminate red from green.
     spec["supported_inference_instance_types"] = [_FIXTURE_DEFAULT_INSTANCE_TYPE] + [
-        t
-        for t in spec.get("supported_inference_instance_types", [])
-        if t != "ml.m5.large"
+        t for t in spec.get("supported_inference_instance_types", []) if t != "ml.m5.large"
     ]
     return spec
 
@@ -50,7 +48,6 @@ def _get_fixture_spec(*args, **kwargs):
     return JumpStartModelSpecs(_load_fixture_spec_dict())
 
 
-
 def _mock_session():
     """Mock sagemaker session following the package's shared-setup convention."""
     mock_session = Mock()
@@ -122,14 +119,14 @@ def spy(self):
         with patch.object(ModelBuilder, "_get_default_instance_type", spy):
             _build_jumpstart_builder()
 
-        self.assertTrue(seen.get("has_region"), "self.region missing during instance-type detection")
+        self.assertTrue(
+            seen.get("has_region"), "self.region missing during instance-type detection"
+        )
         self.assertEqual(seen.get("region"), "us-west-2")
 
     def test_user_provided_instance_type_wins_and_skips_lookup(self):
         """Explicit instance_type must be honored without consulting the spec."""
-        builder, get_deploy_kwargs_mock = _build_jumpstart_builder(
-            instance_type="ml.p5.48xlarge"
-        )
+        builder, get_deploy_kwargs_mock = _build_jumpstart_builder(instance_type="ml.p5.48xlarge")
 
         self.assertEqual(builder.instance_type, "ml.p5.48xlarge")
         self.assertTrue(builder._user_provided_instance_type)
@@ -157,9 +154,9 @@ def test_deploy_passes_spec_default_to_internal_deploy(self):
         builder, _ = _build_jumpstart_builder()
         builder.built_model = Mock()
 
-        with patch.object(ModelBuilder, "_is_model_customization", return_value=False), patch.object(
-            ModelBuilder, "_deploy", return_value=Mock()
-        ) as mock_deploy:
+        with patch.object(
+            ModelBuilder, "_is_model_customization", return_value=False
+        ), patch.object(ModelBuilder, "_deploy", return_value=Mock()) as mock_deploy:
             builder.deploy(endpoint_name="test-endpoint")
 
         mock_deploy.assert_called_once()
@@ -265,9 +262,9 @@ def test_deploy_passes_user_override_to_internal_deploy(self):
         builder, _ = _build_jumpstart_builder(instance_type=self.USER_INSTANCE_TYPE)
         builder.built_model = Mock()
 
-        with patch.object(ModelBuilder, "_is_model_customization", return_value=False), patch.object(
-            ModelBuilder, "_deploy", return_value=Mock()
-        ) as mock_deploy:
+        with patch.object(
+            ModelBuilder, "_is_model_customization", return_value=False
+        ), patch.object(ModelBuilder, "_deploy", return_value=Mock()) as mock_deploy:
             builder.deploy(endpoint_name="test-endpoint")
 
         mock_deploy.assert_called_once()