Merge branch 'master' into dependabot-apache-airflow

Author: zhaoqizqwang

.github/workflows/codebuild-ci-health.yml

@@ -26,7 +26,7 @@ jobs:
     strategy:
         fail-fast: false
         matrix:
-          python-version: ["py38", "py39", "py310"]
+          python-version: ["py38", "py39", "py310", "py311"]
     steps:
       - name: Configure AWS Credentials
         uses: aws-actions/configure-aws-credentials@v4

.github/workflows/codebuild-ci.yml

@@ -29,7 +29,7 @@ jobs:
     strategy:
         fail-fast: false
         matrix:
-          python-version: ["py38", "py39", "py310"]
+          python-version: ["py38","py39","py310","py311"]
     steps:
       - name: Configure AWS Credentials
         uses: aws-actions/configure-aws-credentials@v4

CHANGELOG.md

@@ -1,5 +1,32 @@
 # Changelog
 
+## v2.214.1 (2024-03-27)
+
+### Bug Fixes and Other Changes
+
+ * Update schema dependency version
+ * remove failing deprecated tests from suite
+ * update readme, trigger p311 tests
+ * JumpStart list models flaky tests
+ * fix badge in README
+
+## v2.214.0 (2024-03-22)
+
+### Features
+
+ * add support to ``clarify.py`` for time series explainability jobs
+
+### Bug Fixes and Other Changes
+
+ * remove pytorch test for deprecated function
+ * skip test_experiment_analytics to unblock release
+ * Create workflow module scoped sagemaker_session to resolve test race condition
+ * Simplify how we process test dependencies, which are supposed to include all extras.
+ * skip failing feature store search integ test
+ * skip failing pt test
+ * list jumpstart models with invalid version strings
+ * urge customers to install latest version
+
 ## v2.213.0 (2024-03-15)
 
 ### Features

README.rst

@@ -22,8 +22,8 @@ SageMaker Python SDK
    :target: https://sagemaker.readthedocs.io/en/stable/
    :alt: Documentation Status
 
-.. image:: https://github.com/benieric/sagemaker-python-sdk/actions/workflows/codebuild-ci-health.yml/badge.svg
-    :target: https://github.com/benieric/sagemaker-python-sdk/actions/workflows/codebuild-ci-health.yml
+.. image:: https://github.com/aws/sagemaker-python-sdk/actions/workflows/codebuild-ci-health.yml/badge.svg
+    :target: https://github.com/aws/sagemaker-python-sdk/actions/workflows/codebuild-ci-health.yml
     :alt: CI Health
 
 SageMaker Python SDK is an open source library for training and deploying machine learning models on Amazon SageMaker.
@@ -93,6 +93,7 @@ SageMaker Python SDK is tested on:
 - Python 3.8
 - Python 3.9
 - Python 3.10
+- Python 3.11
 
 AWS Permissions
 ~~~~~~~~~~~~~~~

VERSION

@@ -1 +1 @@
-2.213.1.dev0
+2.214.2.dev0

requirements/extras/test_requirements.txt

@@ -23,20 +23,16 @@ pyvis==0.2.1
 pandas>=1.3.5,<1.5
 scikit-learn==1.3.0
 cloudpickle==2.2.1
-scipy==1.10.1
-urllib3>=1.26.8,<3.0.0
-docker>=5.0.2,<7.0.0
 PyYAML==6.0
-pyspark==3.3.1
-sagemaker-feature-store-pyspark-3.3
 # TODO find workaround
 xgboost>=1.6.2,<=1.7.6
 pillow>=10.0.1,<=11
 transformers>=4.36.0
 sentencepiece==0.1.99
 # https://github.com/triton-inference-server/server/issues/6246
 tritonclient[http]<2.37.0
-onnx==1.14.1
+onnx>=1.15.0
 # tf2onnx==1.15.1
 nbformat>=5.9,<6
 accelerate>=0.24.1,<=0.27.0
+schema==0.7.5

setup.py

@@ -85,8 +85,10 @@ def read_requirements(filename):
 extras["all"] = [item for group in extras.values() for item in group]
 # Tests specific dependencies (do not need to be included in 'all')
 test_dependencies = read_requirements("requirements/extras/test_requirements.txt")
-# remove torch and torchvision if python version is not 3.10
-if sys.version_info.minor != 10:
+# test dependencies are a superset of testing and extra dependencies
+test_dependencies.extend(extras["all"])
+# remove torch and torchvision if python version is not 3.10/3.11
+if sys.version_info.minor != 10 or sys.version_info.minor != 11:
     test_dependencies = [
         module
         for module in test_dependencies
@@ -124,6 +126,7 @@ def read_requirements(filename):
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
     ],
     install_requires=required_packages,
     extras_require=extras,