From f4d894af2ebb66f4492506ebb121509ef8cb8aa3 Mon Sep 17 00:00:00 2001 From: Eduardo Apolinario <653394+eapolinario@users.noreply.github.com> Date: Thu, 4 Apr 2024 13:20:12 -0700 Subject: [PATCH] Set a minimum version to mlflow in flytekit plugin (#2324) * Set `mlflow>=2.10.0` Signed-off-by: Eduardo Apolinario * Run mlflow tests on python 3.11. Signed-off-by: Eduardo Apolinario --------- Signed-off-by: Eduardo Apolinario Co-authored-by: Eduardo Apolinario --- .github/workflows/pythonbuild.yml | 4 ---- plugins/flytekit-mlflow/setup.py | 4 ++-- 2 files changed, 2 insertions(+), 6 deletions(-) diff --git a/.github/workflows/pythonbuild.yml b/.github/workflows/pythonbuild.yml index 7b308ad634..c2e5d96544 100644 --- 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 index 666aff4316..8074f4ed06 100644 --- 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",