From 7a652bc39d72490fea7341f52847406a98ffb9af Mon Sep 17 00:00:00 2001
From: JackZ-db <jack.zhang@databricks.com>
Date: Mon, 24 Jun 2024 17:15:28 -0700
Subject: [PATCH] bumping mlflow to 2.14.1

---
 setup.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index cbffa0b79c..9129b22492 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',
 ]