resolve conflict dependency with numpy 2.0

Author: Roja Reddy Sareddy

requirements/extras/test_requirements.txt

@@ -23,7 +23,7 @@ sagemaker-experiments==0.1.35
 Jinja2==3.1.6
 pyvis==0.2.1
 pandas==2.2.3
-scikit-learn==1.4.0
+scikit-learn==1.6.1
 cloudpickle==2.2.1
 jsonpickle<4.0.0
 PyYAML>=6.0.1

tests/data/serve_resources/mlflow/pytorch/conda.yaml

@@ -20,5 +20,5 @@ dependencies:
   - torch>=2.6.0
   - torchvision>=0.17.0
   - tqdm==4.66.2
-  - scikit-learn==1.3.2
+  - scikit-learn==1.6.1
 name: mlflow-env

tests/data/serve_resources/mlflow/xgboost/conda.yaml

@@ -9,7 +9,7 @@ dependencies:
   - numpy>=2.0.0,<2.3.0
   - pandas==2.2.1
   - psutil==5.9.8
-  - scikit-learn==1.3.2
+  - scikit-learn==1.6.1
   - scipy==1.11.3
   - xgboost==1.7.1
 name: mlflow-env

tests/data/serve_resources/mlflow/xgboost/requirements.txt

@@ -3,6 +3,6 @@ lz4==4.3.2
 numpy>=2.0.0,<2.3.0
 pandas==2.0.3
 psutil==5.9.8
-scikit-learn==1.3.2
+scikit-learn==1.6.1
 scipy==1.11.3
 xgboost==1.7.1