Merge branch 'dev' into 4gpu-connect-test

mosaicml · Jun 25, 2024 · 8fd3fa4 · 8fd3fa4
2 parents f5fe4a9 + 9a6780b
commit 8fd3fa4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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',
 ]