diff --git a/README.rst b/README.rst index 4c1d377..b333963 100644 --- a/README.rst +++ b/README.rst @@ -2,13 +2,16 @@ SageMaker Scikit-learn Container =========================== +⚠️ Moved — use ``aws/deep-learning-containers`` instead +======================================================= + .. important:: **This repository is no longer actively maintained for new scikit-learn version upgrades.** All future scikit-learn version upgrades and container releases are tracked in the `AWS Deep Learning Containers `__ repository - under `docker/sklearn `__. + under `docker/sklearn `__. This repository remains available as a source reference for the SageMaker scikit-learn serving framework code. diff --git a/requirements.txt b/requirements.txt index f12fea3..a2be06a 100644 --- a/requirements.txt +++ b/requirements.txt @@ -1,30 +1,30 @@ -numpy==2.1.0 -scikit-learn==1.4.2 -scipy>=1.9.0 -pandas==2.3.2 -pyarrow==17.0.0 -boto3==1.28.57 -botocore>=1.31.57,<1.32.0 -certifi -cryptography==46.0.7 -Flask==1.1.1 -gevent==23.9.1 -gunicorn==23.0.0 -itsdangerous==2.0.1 -jinja2<3.0 -MarkupSafe<2.0 +numpy==2.5.1 +scikit-learn==1.9.0 +scipy==1.18.0 +narwhals>=2.0.1 +joblib>=1.4.0 +threadpoolctl>=3.5.0 +pandas==2.2.3 +pyarrow==25.0.0 +boto3==1.43.46 +botocore>=1.43.46,<1.44 +certifi==2026.6.17 +cryptography==49.0.0 +Flask==3.1.3 +gevent==26.5.0 +gunicorn==26.0.0 model-archiver==1.0.3 -multi-model-server==1.1.1 +multi-model-server==1.1.2 protobuf==3.20.2 -psutil==5.7.2 -python-dateutil -PyYAML==6.0.1 +psutil==7.2.2 +python-dateutil==2.9.0.post0 +PyYAML==6.0.3 retrying==1.3.3 sagemaker-containers==2.8.6.post2 -sagemaker-inference==1.2.0 +sagemaker-inference==1.5.5 sagemaker-training==4.8.0 -setuptools==80.9.0 -six==1.16.0 -urllib3==1.26.17 -Werkzeug==2.0.3 -wheel==0.45.1 +setuptools>=80.9.0,<81 +six>=1.16.0,<2 +urllib3==2.7.0 +Werkzeug==3.1.8 +wheel==0.47.0 diff --git a/test/unit/test_modules.py b/test/unit/test_modules.py index f56053f..6f98e83 100644 --- a/test/unit/test_modules.py +++ b/test/unit/test_modules.py @@ -4,7 +4,7 @@ def test_pandas_version(): import pandas as pd major, minor, patch = pd.__version__.split('.') - assert major == '2' + assert int(major) >= 2 def test_pyarrow_to_parquet_conversion_regression_issue_106():