Set a minimum version to mlflow in flytekit plugin (#2324)

* Set `mlflow>=2.10.0` Signed-off-by: Eduardo Apolinario <[email protected]> * Run mlflow tests on python 3.11. Signed-off-by: Eduardo Apolinario <[email protected]> --------- Signed-off-by: Eduardo Apolinario <[email protected]> Co-authored-by: Eduardo Apolinario <[email protected]>
flyteorg · Apr 4, 2024 · f4d894a · f4d894a
1 parent 0a7a72a
commit f4d894a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 6 deletions.
diff --git a/.github/workflows/pythonbuild.yml b/.github/workflows/pythonbuild.yml
@@ -350,10 +350,6 @@ jobs:
             plugin-names: "flytekit-onnx-scikitlearn"
           - python-version: 3.11
             plugin-names: "flytekit-onnx-tensorflow"
-          # numba, a dependency of mlflow, doesn't support python 3.11
-          # https://github.com/numba/numba/issues/8304
-          - python-version: 3.11
-            plugin-names: "flytekit-mlflow"
           # vaex currently doesn't support python 3.11
           - python-version: 3.11
             plugin-names: "flytekit-vaex"

diff --git a/plugins/flytekit-mlflow/setup.py b/plugins/flytekit-mlflow/setup.py
@@ -4,8 +4,7 @@
 
 microlib_name = f"flytekitplugins-{PLUGIN_NAME}"
 
-# TODO: support mlflow 2.0+
-plugin_requires = ["flytekit>=1.1.0,<2.0.0", "plotly", "mlflow<2.0.0", "pandas"]
+plugin_requires = ["flytekit>=1.1.0,<2.0.0", "plotly", "mlflow>=2.10.0", "pandas"]
 
 __version__ = "0.0.0+develop"
 
@@ -27,6 +26,7 @@
         "Programming Language :: Python :: 3.8",
         "Programming Language :: Python :: 3.9",
         "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
         "Topic :: Scientific/Engineering",
         "Topic :: Scientific/Engineering :: Artificial Intelligence",
         "Topic :: Software Development",