diff --git a/requirements/extras/test_requirements.txt b/requirements/extras/test_requirements.txt index 60904c51b0..63e9e4b477 100644 --- a/requirements/extras/test_requirements.txt +++ b/requirements/extras/test_requirements.txt @@ -37,4 +37,4 @@ nbformat>=5.9,<6 accelerate>=0.24.1,<=0.27.0 schema==0.7.5 tensorflow>=2.1,<=2.16 -mlflow>=2.12.2,<2.13 +mlflow==2.13.2 diff --git a/tests/data/serve_resources/mlflow/xgboost/requirements.txt b/tests/data/serve_resources/mlflow/xgboost/requirements.txt index 1130dcaec5..50a156469b 100644 --- a/tests/data/serve_resources/mlflow/xgboost/requirements.txt +++ b/tests/data/serve_resources/mlflow/xgboost/requirements.txt @@ -3,6 +3,6 @@ lz4==4.3.2 numpy==1.24.4 pandas==2.0.3 psutil==5.9.8 -scikit-learn==1.3.2 +scikit-learn==1.5.0 scipy==1.10.1 xgboost==1.7.1