diff --git a/setup.cfg b/setup.cfg
index 5043a5069f..20cdb083d4 100644
--- a/setup.cfg
+++ b/setup.cfg
@@ -37,6 +37,8 @@ install_requires =
 [options.extras_require]
 HE =
     tenseal==0.3.0
+mlflow =
+    mlflow >= 2.0.1
 
 [options.entry_points]
 console_scripts =