Merge branch 'master' into attrs-update

nargokul · web-flow · commit 6e80a03ee309 · 2025-06-03T10:03:45.000-07:00
diff --git a/CHANGELOG.md b/CHANGELOG.md
@@ -1,5 +1,18 @@
 # Changelog
 
+## v2.245.0 (2025-05-28)
+
+### Features
+
+ * Correct mypy type checking through PEP 561
+
+### Bug Fixes and Other Changes
+
+ * MLFLow update for dependabot
+ * addWaiterTimeoutHandling
+ * merge method inputs with class inputs
+ * update image_uri_configs 05-20-2025 07:18:17 PST
+
 ## v2.244.2 (2025-05-19)
 
 ### Bug Fixes and Other Changes
diff --git a/VERSION b/VERSION
@@ -1 +1 @@
-2.244.3.dev0
+2.245.1.dev0
diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt
@@ -44,7 +44,7 @@ nbformat>=5.9,<6
 accelerate>=0.24.1,<=0.27.0
 schema==0.7.5
 tensorflow>=2.16.2,<=2.18.0
-mlflow>=2.12.2,<3
+mlflow>=2.14.2,<3
 huggingface_hub==0.26.2
 uvicorn>=0.30.1
 fastapi==0.115.4
diff --git a/src/sagemaker/estimator.py b/src/sagemaker/estimator.py
@@ -456,6 +456,9 @@ def __init__(
                 A dictionary containing the hyperparameters to
                 initialize this estimator with. (Default: None).
 
+                If a source directory is specified, the set_hyperparameters method escapes
+                the dict argument as JSON, and updates the private hyperparameter attribute.
+
                 .. caution::
                     You must not include any security-sensitive information, such as
                     account access IDs, secrets, and tokens, in the dictionary for configuring
diff --git a/src/sagemaker/image_uri_config/sagemaker-tritonserver.json b/src/sagemaker/image_uri_config/sagemaker-tritonserver.json
diff --git a/tests/unit/sagemaker/mlflow/test_forward_sagemaker_metrics.py b/tests/unit/sagemaker/mlflow/test_forward_sagemaker_metrics.py
