diff --git a/python/requirements/ml/requirements_train.txt b/python/requirements/ml/requirements_train.txt index 5f999b3648f4..a28d2333b6b7 100644 --- a/python/requirements/ml/requirements_train.txt +++ b/python/requirements/ml/requirements_train.txt @@ -3,7 +3,7 @@ -r requirements_dl.txt mlflow==1.30.0; python_version <= '3.7' -mlflow==2.2.2; python_version > '3.7' +mlflow==2.16.0; python_version > '3.7' tensorboardX==2.4.1 # Dependencies for Hugging Face examples & tests: diff --git a/python/requirements/ml/requirements_tune.txt b/python/requirements/ml/requirements_tune.txt index f85c4d9299f5..42e023fe473f 100644 --- a/python/requirements/ml/requirements_tune.txt +++ b/python/requirements/ml/requirements_tune.txt @@ -23,7 +23,7 @@ jupyterlab==3.6.1 lightgbm==3.3.5 matplotlib!=3.4.3 mlflow==1.30.0; python_version <= '3.7' -mlflow==2.2.2; python_version > '3.7' +mlflow==2.16.0; python_version > '3.7' # Unavailable for arm64 in more recent versions mxnet==1.8.0.post0; platform_machine != "arm64" nevergrad==0.4.3.post7