Avoid installing the latest MLfow to prevent doctests from failing (#…

…3135) * Avoid installing latest mlflow
optuna · Nov 30, 2021 · 2a9f863 · 2a9f863
1 parent 51b5483
commit 2a9f863
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/setup.py b/setup.py
@@ -61,7 +61,7 @@ def get_extras_require() -> Dict[str, List[str]]:
             "plotly>=4.0.0",
             "scikit-learn>=0.24.2",
             "scikit-optimize",
-            "mlflow",
+            "mlflow<1.22.0",
         ],
         "document": [
             # TODO(nzw): Remove the version constraint after resolving the issue
@@ -89,7 +89,7 @@ def get_extras_require() -> Dict[str, List[str]]:
             "fakeredis",
             "lightgbm",
             "matplotlib>=3.0.0",
-            "mlflow",
+            "mlflow<1.22.0",
             "mpi4py",
             "mxnet",
             "pandas",
@@ -131,7 +131,7 @@ def get_extras_require() -> Dict[str, List[str]]:
             "chainer>=5.0.0",
             "cma",
             "lightgbm",
-            "mlflow",
+            "mlflow<1.22.0",
             "wandb",
             "mpi4py",
             "mxnet",