Set range to mlflow package version

shixiaoyu321 · Nov 23, 2022 · 282b16b · 282b16b
1 parent fe77c8b
commit 282b16b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/deploy/mlflow-triton-plugin/setup.py b/deploy/mlflow-triton-plugin/setup.py
@@ -32,6 +32,6 @@
     long_description=open("README.md").read(),
     long_description_content_type="text/markdown",
     packages=find_packages(),
-    install_requires=["mlflow>=1.15.0", "tritonclient[all]"],
+    install_requires=["mlflow>=1.15.0,<2.0", "tritonclient[all]"],
     entry_points={"mlflow.deployments": "triton=mlflow_triton.deployments"},
 )