fix(jumpstart): remove core dependency on serve shims

Use the canonical core classes that the deprecated serve shims export. This keeps JumpStart factory imports available in core-only environments.

---
X-AI-Prompt: Can you fix the pull request hooks and keep the new boundary tests in sagemaker-core?
X-AI-Tool: claude-code

Author: evakravi

sagemaker-core/src/sagemaker/core/jumpstart/factory/utils.py

@@ -23,7 +23,7 @@
     model_uris,
     script_uris,
 )
-from sagemaker.serve.async_inference.async_inference_config import AsyncInferenceConfig
+from sagemaker.core.inference_config import AsyncInferenceConfig, ServerlessInferenceConfig
 from sagemaker.core.deserializers.base import BaseDeserializer
 from sagemaker.core.serializers.base import BaseSerializer
 from sagemaker.core.explainer.explainer_config import ExplainerConfig
@@ -74,7 +74,6 @@
 
 from sagemaker.core.model_monitor.data_capture_config import DataCaptureConfig
 
-from sagemaker.serve.serverless.serverless_inference_config import ServerlessInferenceConfig
 from sagemaker.core.helper.session_helper import Session
 from sagemaker.core.common_utils import (
     camel_case_to_pascal_case,
@@ -83,11 +82,10 @@
     Tags,
 )
 from sagemaker.core.helper.pipeline_variable import PipelineVariable
-from sagemaker.serve.compute_resource_requirements.resource_requirements import ResourceRequirements
+from sagemaker.core.resource_requirements import ResourceRequirements
 from sagemaker.core import resource_requirements
 from sagemaker.core.enums import EndpointType
 
-
 KwargsType = Union[
     JumpStartModelDeployKwargs,
     JumpStartModelInitKwargs,
@@ -551,9 +549,7 @@ def _add_additional_model_data_sources_to_kwargs(
             provider=data_source.provider, region=kwargs.region
         )
         if content_bucket is None:
-            raise ValueError(
-                f"Unable to get content bucket for Neo in {kwargs.region} region."
-            )
+            raise ValueError(f"Unable to get content bucket for Neo in {kwargs.region} region.")
         data_source.s3_data_source.set_bucket(content_bucket)
     api_shape_additional_model_data_sources = (
         [

sagemaker-core/src/sagemaker/core/jumpstart/utils.py

@@ -11,6 +11,7 @@
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
 """This module contains utils for JumpStart."""
+
 from __future__ import absolute_import
 
 from typing import Optional

sagemaker-core/tests/integ/jumpstart/test_bucket_utils.py

@@ -11,6 +11,7 @@
 # ANY KIND, either express or implied. See the License for the specific
 # language governing permissions and limitations under the License.
 """Regression tests for JumpStart metadata setup in EU Sovereign Cloud."""
+
 from __future__ import absolute_import
 
 import boto3

sagemaker-core/tests/unit/jumpstart/test_factory_utils.py

@@ -68,9 +68,7 @@ def test_jumpstart_draft_model_bypasses_neo(
         self, mock_get_neo_bucket, mock_get_jumpstart_bucket
     ):
         mock_get_jumpstart_bucket.return_value = "jumpstart-bucket"
-        data_source = self._draft_model_source(
-            {"name": "JumpStart", "classification": "ungated"}
-        )
+        data_source = self._draft_model_source({"name": "JumpStart", "classification": "ungated"})
         kwargs = Mock(region="eusc-de-east-1", additional_model_data_sources=None)
         kwargs.specs.get_speculative_decoding_s3_data_sources.return_value = [data_source]
 

sagemaker-core/tests/unit/test_jumpstart_utils.py

@@ -1583,7 +1583,9 @@ def test_add_instance_rate_stats_none_metrics(self):
         result = utils.add_instance_rate_stats_to_benchmark_metrics("us-west-2", None)
         assert result is None
 
-    @pytest.mark.skip(reason="Requires AWS Pricing API permissions which are not available in CI environment")
+    @pytest.mark.skip(
+        reason="Requires AWS Pricing API permissions which are not available in CI environment"
+    )
     @patch("sagemaker.core.common_utils.get_instance_rate_per_hour")
     def test_add_instance_rate_stats_success(self, mock_get_rate):
         """Test successfully adding instance rate stats"""
@@ -1905,6 +1907,7 @@ def test_get_draft_model_content_bucket_other_provider(self, mock_neo):
         result = utils.get_draft_model_content_bucket(provider, "us-west-2")
         assert result == "neo-bucket"
 
+
 class TestRemoveEnvVarFromEstimatorKwargsIfAcceptEulaPresent:
     """Test cases for remove_env_var_from_estimator_kwargs_if_accept_eula_present function"""