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",