diff --git a/requirements.txt b/requirements.txt
index 538e97c..514fb96 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,2 +1,2 @@
-mlflow==0.8.0
+mlflow==2.3.1
 yacs==0.1.4
diff --git a/setup.py b/setup.py
index b2c2c08..2eff95e 100644
--- a/setup.py
+++ b/setup.py
@@ -12,7 +12,7 @@
     description='My machine learning experimentation workflow',
     long_description=open('README.txt').read(),
     install_requires=[
-        "mlflow == 0.8.0",
+        "mlflow == 2.3.1",
         "yacs == 0.1.4",
     ],
 )
\ No newline at end of file