Merge branch 'master' into feat/feature-store-fp-lf

Author: BassemHalim

CHANGELOG.md

@@ -1,4 +1,14 @@
 # Changelog
+## v3.10.1 (2026-05-07)
+
+### Bug Fixes
+- Fix KMS key propagation in check steps (QualityCheckStep, ClarifyCheckStep)
+- Fix JumpStart network isolation in ModelBuilder
+- Fix base_model_arn construction to use private hub when SAGEMAKER_HUB_NAME is set
+- Fix imports for Model Customization interfaces
+- Fix handling of unrecognized JumpStart container images in ModelBuilder
+- Increase default timeout for training jobs
+
 ## v3.10.0 (2026-05-01)
 
 ### New Features

VERSION

@@ -1 +1 @@
-3.10.0
+3.10.1

pyproject.toml

@@ -33,10 +33,10 @@ classifiers = [
   "Programming Language :: Python :: 3.12",
 ]
 dependencies = [
-  "sagemaker-core>=2.10.0,<3.0.0",
-  "sagemaker-train>=1.10.0,<2.0.0",
-  "sagemaker-serve>=1.10.0,<2.0.0",
-  "sagemaker-mlops>=1.10.0,<2.0.0",
+  "sagemaker-core>=2.10.1,<3.0.0",
+  "sagemaker-train>=1.10.1,<2.0.0",
+  "sagemaker-serve>=1.10.1,<2.0.0",
+  "sagemaker-mlops>=1.10.1,<2.0.0",
 ]
 
 [project.optional-dependencies]

sagemaker-core/CHANGELOG.md

@@ -1,4 +1,10 @@
 # Changelog
+## v2.10.1 (2026-05-07)
+
+### Bug Fixes
+
+- Fix image retriever tests
+
 ## v2.10.0 (2026-05-01)
 
 ### New Features

sagemaker-core/VERSION

@@ -1 +1 @@
-2.10.0
+2.10.1

sagemaker-core/src/sagemaker/core/image_retriever/image_retriever.py

@@ -52,6 +52,8 @@
 
 
 class ImageRetriever:
+    _config = SageMakerConfig()
+
     @staticmethod
     def retrieve_hugging_face_uri(
         region: str,
@@ -110,7 +112,7 @@ def retrieve_hugging_face_uri(
         args = dict(locals())
         for name, val in args.items():
             if name in CONFIGURABLE_ATTRIBUTES and not val:
-                default_value = SageMakerConfig.resolve_value_from_config(
+                default_value = ImageRetriever._config.resolve_value_from_config(
                     config_path=_simple_path(
                         SAGEMAKER, MODULES, IMAGE_RETRIEVER, to_camel_case(name)
                     )
@@ -499,7 +501,7 @@ def retrieve(
         args = dict(locals())
         for name, val in args.items():
             if name in CONFIGURABLE_ATTRIBUTES and not val:
-                default_value = SageMakerConfig.resolve_value_from_config(
+                default_value = ImageRetriever._config.resolve_value_from_config(
                     config_path=_simple_path(
                         SAGEMAKER, MODULES, IMAGE_RETRIEVER, to_camel_case(name)
                     )

sagemaker-core/src/sagemaker/core/image_retriever/image_retriever_utils.py

@@ -28,6 +28,7 @@
     GRAVITON_ALLOWED_FRAMEWORKS,
 )
 from sagemaker.core.common_utils import _botocore_resolver, get_instance_type_family
+from sagemaker.core.spark import defaults
 
 logger = logging.getLogger(__name__)
 

sagemaker-core/tests/integ/image_retriever/test_image_retriever.py

@@ -12,7 +12,6 @@
 from sagemaker.core.config.config_manager import SageMakerConfig
 
 
-@pytest.mark.skip("Disabling this for now, Need to be fixed")
 @pytest.mark.integ
 def test_retrieve_image_uri():
     image_uri = ImageRetriever.retrieve("clarify", "us-west-2")
@@ -28,7 +27,7 @@ def test_retrieve_image_uri():
     )
     assert (
         image_uri
-        == "053634841547.dkr.ecr.us-west-1.amazonaws.com/sagemaker-distribution-prod:3.0.0-gpu"
+        == "053634841547.dkr.ecr.us-west-1.amazonaws.com/sagemaker-distribution-prod:3.2.0-gpu"
     )
 
     image_uri = ImageRetriever.retrieve(
@@ -56,7 +55,6 @@ def test_retrieve_image_uri():
     )
 
 
-@pytest.mark.skip("Disabling this for now, Need to be fixed")
 @pytest.mark.integ
 def test_retrieve_pytorch_uri():
     image_uri = ImageRetriever.retrieve_pytorch_uri(
@@ -72,7 +70,6 @@ def test_retrieve_pytorch_uri():
     )
 
 
-@pytest.mark.skip("Disabling this for now, Need to be fixed")
 @pytest.mark.integ
 def test_retrieve_hugging_face_uri():
     image_uri = ImageRetriever.retrieve_hugging_face_uri(
@@ -84,22 +81,23 @@ def test_retrieve_hugging_face_uri():
         base_framework_version="pytorch2.0.0",
         container_version="cu110-ubuntu20.04",
     )
-    assert image_uri == "763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-training"
-    ":2.0.0-transformers4.28.1-gpu-py310-cu118-ubuntu20.04"
+    assert (
+        image_uri
+        == "763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-training"
+        ":2.0.0-transformers4.28.1-gpu-py310-cu118-ubuntu20.04"
+    )
 
 
-@pytest.mark.skip("Disabling this for now, Need to be fixed")
 @pytest.mark.integ
 def test_retrieve_base_python_image_uri():
-    image_uri = ImageRetriever.retrieve_base_python_image_uri()
+    image_uri = ImageRetriever.retrieve_base_python_image_uri(region="us-west-2")
     assert image_uri == "236514542706.dkr.ecr.us-west-2.amazonaws.com/sagemaker-base-python-310:1.0"
 
 
-@pytest.mark.skip("Disabling this for now, Need to be fixed")
-@pytest.mark.integ
+@pytest.mark.skip(reason="Test is failing due to locals()[name] = default_value in Image Retriever")
 @patch.object(SageMakerConfig, "resolve_value_from_config")
 def test_retrieve_image_uri_intelligent_default(mock_load_config):
-    def custom_return(config_path):
+    def custom_return(config_path=None, **kwargs):
         if config_path == _simple_path(
             SAGEMAKER, PYTHON_SDK, MODULES, IMAGE_RETRIEVER, "ImageScope"
         ):

sagemaker-mlops/CHANGELOG.md

@@ -1,4 +1,10 @@
 # Changelog
+## v1.10.1 (2026-05-07)
+
+### Bug Fixes
+
+- Fix KMS key propagation in QualityCheckStep and ClarifyCheckStep
+
 ## v1.10.0 (2026-05-01)
 
 ### New Features

sagemaker-mlops/VERSION

@@ -1 +1 @@
-1.10.0
+1.10.1