feat: bump scikit-learn to 1.9.0 and modernize transitive deps

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 <https://github.com/aws/deep-learning-containers>`__ repository
-   under `docker/sklearn <https://github.com/aws/deep-learning-containers/tree/master/docker/sklearn>`__.
+   under `docker/sklearn <https://github.com/aws/deep-learning-containers/tree/main/docker/sklearn>`__.
 
    This repository remains available as a source reference for the SageMaker scikit-learn serving framework code.
 

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

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():