diff --git a/python/batch-predictor/requirements.txt b/python/batch-predictor/requirements.txt
index 308bad581..92077f271 100644
--- a/python/batch-predictor/requirements.txt
+++ b/python/batch-predictor/requirements.txt
@@ -1,7 +1,7 @@
 cloudpickle==2.0.0
 findspark==2.0.1
 merlin-sdk==0.44.0
-mlflow==1.26.1
+mlflow==2.17.2
 pyarrow>=0.14.1,<=17.0.0
 pyspark==3.0.1
 setuptools<75
\ No newline at end of file