setup mlflow as optional requirements

NVIDIA · Jan 16, 2023 · 12676dd · 12676dd
1 parent 97c9c27
commit 12676dd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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 =