diff --git a/setup.py b/setup.py index 0afe4ed314..a1deea27c7 100644 --- a/setup.py +++ b/setup.py @@ -223,7 +223,7 @@ def package_files(prefix: str, directory: str, extension: str): ] extra_deps['mlflow'] = [ - 'mlflow>=2.11.1,<3.0', + 'mlflow>=2.14.1,<3.0', 'databricks-sdk==0.28.0', 'pynvml>=11.5.0,<12', ]