feat: update mlflow version to support newer version of protobuf

caraml-dev · Sep 20, 2023 · 592dc85 · 592dc85
1 parent 072325c
commit 592dc85
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/python/sdk/setup.py b/python/sdk/setup.py
@@ -30,8 +30,7 @@
     "cookiecutter>=1.7.2",
     "docker>=4.2.1",
     "google-cloud-storage>=1.19.0",
-    "mlflow>=1.2.0,<=1.23.0", #  for py3.11 due to proto -> "mlflow>=1.26.1",
-    "protobuf>=3.0.0,<4.0.0", #  for py3.11 due to proto -> "protobuf>=4.0.0,<5.0dev",
+    "mlflow>=1.26.1,<2.0.0",
     "PyPrind>=2.11.2",
     "python_dateutil>=2.5.3",
     "PyYAML>=5.4",