fix versions of mlflow and protobuf to fix test breakage

databrickslabs · May 20, 2022 · bf56196 · bf56196
1 parent 3f4ef4f
commit bf56196
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -11,7 +11,8 @@
     "click>=7.1.2",
     "retry>=0.9.2",
     "requests>=2.24.0",
-    "mlflow>=1.23.0",
+    "mlflow==1.26.0",
+    "protobuf==3.20.1",
     "tqdm>=4.50.0",
     "azure-identity>=1.7.1",
     "azure-mgmt-datafactory>=2.2.0",