diff --git a/requirements.txt b/requirements.txt
index deab1c8..d903d04 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -8,7 +8,7 @@ torchvision==0.6.0
 torch==1.5.0
 torch_geometric==1.4.3
 numpy==1.18.1
-mlflow==1.8.0
+mlflow==2.3.1
 opencv_python==4.2.0.34
 pytorch_lightning==0.7.6
 Pillow==8.2.0