Use scikit-learn 1.8.0
diff --git a/sdks/python/apache_beam/examples/inference/sklearn_examples_requirements.txt b/sdks/python/apache_beam/examples/inference/sklearn_examples_requirements.txt index 30dbdb2..a0ec06b 100644 --- a/sdks/python/apache_beam/examples/inference/sklearn_examples_requirements.txt +++ b/sdks/python/apache_beam/examples/inference/sklearn_examples_requirements.txt
@@ -20,4 +20,4 @@ # However, newer sklearn is needed for testing on newer Python version scikit-learn==1.0.2; python_version < '3.11' # bump sklearn version when new Python version is supported -scikit-learn==1.7.1; python_version >= '3.11' +scikit-learn==1.8.0; python_version >= '3.11'
diff --git a/sdks/python/container/ml/py310/base_image_requirements.txt b/sdks/python/container/ml/py310/base_image_requirements.txt index 802137f..297222c 100644 --- a/sdks/python/container/ml/py310/base_image_requirements.txt +++ b/sdks/python/container/ml/py310/base_image_requirements.txt
@@ -192,7 +192,7 @@ rpds-py==0.30.0 rsa==4.9.1 safetensors==0.7.0 -scikit-learn==1.7.2 +scikit-learn==1.8.0 scipy==1.15.3 scramp==1.4.8 SecretStorage==3.5.0
diff --git a/sdks/python/container/ml/py310/gpu_image_requirements.txt b/sdks/python/container/ml/py310/gpu_image_requirements.txt index 74b3ff5..8bac85fd 100644 --- a/sdks/python/container/ml/py310/gpu_image_requirements.txt +++ b/sdks/python/container/ml/py310/gpu_image_requirements.txt
@@ -256,7 +256,7 @@ rpds-py==0.30.0 rsa==4.9.1 safetensors==0.7.0 -scikit-learn==1.7.2 +scikit-learn==1.8.0 scipy==1.15.3 scramp==1.4.8 SecretStorage==3.5.0
diff --git a/sdks/python/container/ml/py311/base_image_requirements.txt b/sdks/python/container/ml/py311/base_image_requirements.txt index 21ec3c9..b79ae7f 100644 --- a/sdks/python/container/ml/py311/base_image_requirements.txt +++ b/sdks/python/container/ml/py311/base_image_requirements.txt
@@ -190,7 +190,7 @@ rpds-py==0.30.0 rsa==4.9.1 safetensors==0.7.0 -scikit-learn==1.7.2 +scikit-learn==1.8.0 scipy==1.17.0 scramp==1.4.8 SecretStorage==3.5.0
diff --git a/sdks/python/container/ml/py311/gpu_image_requirements.txt b/sdks/python/container/ml/py311/gpu_image_requirements.txt index 60b5e3a..7db849c 100644 --- a/sdks/python/container/ml/py311/gpu_image_requirements.txt +++ b/sdks/python/container/ml/py311/gpu_image_requirements.txt
@@ -254,7 +254,7 @@ rpds-py==0.30.0 rsa==4.9.1 safetensors==0.7.0 -scikit-learn==1.7.2 +scikit-learn==1.8.0 scipy==1.17.0 scramp==1.4.8 SecretStorage==3.5.0
diff --git a/sdks/python/container/py310/base_image_requirements.txt b/sdks/python/container/py310/base_image_requirements.txt index 17cc0dd..920d89e 100644 --- a/sdks/python/container/py310/base_image_requirements.txt +++ b/sdks/python/container/py310/base_image_requirements.txt
@@ -169,7 +169,7 @@ requests-mock==1.12.1 rpds-py==0.30.0 rsa==4.9.1 -scikit-learn==1.7.2 +scikit-learn==1.8.0 scipy==1.15.3 scramp==1.4.8 SecretStorage==3.5.0
diff --git a/sdks/python/container/py311/base_image_requirements.txt b/sdks/python/container/py311/base_image_requirements.txt index 8433ca2..ad0be5d 100644 --- a/sdks/python/container/py311/base_image_requirements.txt +++ b/sdks/python/container/py311/base_image_requirements.txt
@@ -167,7 +167,7 @@ requests-mock==1.12.1 rpds-py==0.30.0 rsa==4.9.1 -scikit-learn==1.7.2 +scikit-learn==1.8.0 scipy==1.17.0 scramp==1.4.8 SecretStorage==3.5.0
diff --git a/sdks/python/setup.py b/sdks/python/setup.py index 4291bba..a255b10 100644 --- a/sdks/python/setup.py +++ b/sdks/python/setup.py
@@ -456,7 +456,7 @@ 'pytest>=7.1.2,<9.0', 'pytest-xdist>=2.5.0,<4', 'pytest-timeout>=2.1.0,<3', - 'scikit-learn>=0.20.0,<1.8.0', + 'scikit-learn>=0.20.0,<=1.8.0', 'sqlalchemy>=1.3,<3.0', 'psycopg2-binary>=2.8.5,<3.0', 'testcontainers[mysql,kafka,milvus]>=4.0.0,<5.0.0',