From 13fb51b477bb86909e786897fbf15c8bb6ee8e13 Mon Sep 17 00:00:00 2001 From: Clayne Robison Date: Fri, 21 Jun 2024 16:30:04 -0700 Subject: [PATCH] Bump scikit-learn version to 1.5.0 (#2292) --- .../tensorflow/wide_deep/inference/requirements.txt | 2 +- .../dependencies/conda/requirements-test.yaml | 2 +- datasets/cloud_data_connector/samples/azure/requirements.txt | 2 +- .../samples/interoperability/requirements.txt | 2 +- docs/notebooks/transfer_learning/requirements.txt | 2 +- .../pytorch/bert_large/training/gpu/requirements.txt | 2 +- models_v2/pytorch/dlrm/inference/gpu/requirements.txt | 2 +- models_v2/pytorch/yolov5/inference/gpu/requirements.txt | 2 +- quickstart/recommendation/pytorch/dlrm/requirements.txt | 2 +- 9 files changed, 9 insertions(+), 9 deletions(-) diff --git a/benchmarks/recommendation/tensorflow/wide_deep/inference/requirements.txt b/benchmarks/recommendation/tensorflow/wide_deep/inference/requirements.txt index fda4dd66e..4ff4232d3 100755 --- a/benchmarks/recommendation/tensorflow/wide_deep/inference/requirements.txt +++ b/benchmarks/recommendation/tensorflow/wide_deep/inference/requirements.txt @@ -1,2 +1,2 @@ pandas -scikit-learn +scikit-learn>=1.5.0 diff --git a/datasets/cloud_data_connector/dependencies/conda/requirements-test.yaml b/datasets/cloud_data_connector/dependencies/conda/requirements-test.yaml index 39f837c04..486c784a9 100644 --- a/datasets/cloud_data_connector/dependencies/conda/requirements-test.yaml +++ b/datasets/cloud_data_connector/dependencies/conda/requirements-test.yaml @@ -30,7 +30,7 @@ dependencies: - pytest-cov ==4.1.0 - pytest-mock ==3.11.1 - pytest-xdist ==3.3.1 - - scikit-learn ==1.2.2 + - scikit-learn >=1.5.0 - tox ==3.27.1 - xlrd ==2.0.1 - pip: diff --git a/datasets/cloud_data_connector/samples/azure/requirements.txt b/datasets/cloud_data_connector/samples/azure/requirements.txt index d8ae6ce2c..861d669e4 100644 --- a/datasets/cloud_data_connector/samples/azure/requirements.txt +++ b/datasets/cloud_data_connector/samples/azure/requirements.txt @@ -1,3 +1,3 @@ mlflow -scikit-learn==1.2.2 +scikit-learn>=1.5.0 xlrd==2.0.1 diff --git a/datasets/cloud_data_connector/samples/interoperability/requirements.txt b/datasets/cloud_data_connector/samples/interoperability/requirements.txt index df06aa0e8..6631a0213 100644 --- a/datasets/cloud_data_connector/samples/interoperability/requirements.txt +++ b/datasets/cloud_data_connector/samples/interoperability/requirements.txt @@ -2,5 +2,5 @@ build==0.10.0 mlflow notebook==6.5.4 pandas-gbq==0.19.1 -scikit-learn==1.2.2 +scikit-learn>=1.5.0 xlrd==2.0.1 diff --git a/docs/notebooks/transfer_learning/requirements.txt b/docs/notebooks/transfer_learning/requirements.txt index 0f67e6fff..a8c539957 100644 --- a/docs/notebooks/transfer_learning/requirements.txt +++ b/docs/notebooks/transfer_learning/requirements.txt @@ -31,5 +31,5 @@ tf-models-official torch==1.13.1 torchvision==0.14.1 transformers~=4.38.0 # upgraded to avoid a vulnerability -urllib3~=2.0.2 +urllib3>=2.2.2 wget~=3.2 diff --git a/models/language_modeling/pytorch/bert_large/training/gpu/requirements.txt b/models/language_modeling/pytorch/bert_large/training/gpu/requirements.txt index bae63c067..8ee941e80 100644 --- a/models/language_modeling/pytorch/bert_large/training/gpu/requirements.txt +++ b/models/language_modeling/pytorch/bert_large/training/gpu/requirements.txt @@ -13,7 +13,7 @@ pandas psutil pyyaml requests -scikit-learn +scikit-learn>=1.5.0 scipy==1.10.0 setuptools six diff --git a/models_v2/pytorch/dlrm/inference/gpu/requirements.txt b/models_v2/pytorch/dlrm/inference/gpu/requirements.txt index c000598e6..2439288cf 100644 --- a/models_v2/pytorch/dlrm/inference/gpu/requirements.txt +++ b/models_v2/pytorch/dlrm/inference/gpu/requirements.txt @@ -2,7 +2,7 @@ future numpy onnx pydot -scikit-learn +scikit-learn>=1.5.0 torchrec-nightly torchx-nightly tqdm diff --git a/models_v2/pytorch/yolov5/inference/gpu/requirements.txt b/models_v2/pytorch/yolov5/inference/gpu/requirements.txt index d1a42842e..5943b36fc 100644 --- a/models_v2/pytorch/yolov5/inference/gpu/requirements.txt +++ b/models_v2/pytorch/yolov5/inference/gpu/requirements.txt @@ -28,7 +28,7 @@ seaborn>=0.11.0 # onnx-simplifier>=0.4.1 # ONNX simplifier # onnx>=1.9.0 # ONNX export # openvino-dev # OpenVINO export -# scikit-learn==0.19.2 # CoreML quantization +# scikit-learn>=1.5.0 # CoreML quantization # tensorflow>=2.4.1 # TFLite export (or tensorflow-cpu, tensorflow-aarch64) # tensorflowjs>=3.9.0 # TF.js export diff --git a/quickstart/recommendation/pytorch/dlrm/requirements.txt b/quickstart/recommendation/pytorch/dlrm/requirements.txt index 1c0eeba8c..1d580317a 100644 --- a/quickstart/recommendation/pytorch/dlrm/requirements.txt +++ b/quickstart/recommendation/pytorch/dlrm/requirements.txt @@ -1,6 +1,6 @@ future numpy -scikit-learn +scikit-learn>=1.5.0 torch tqdm wheel==0.38.0 # upgraded to avoid a vulnerability