Skip to main content

Apache Airflow provider for MLflow

Project description

PyPI Version PyPI - Python Version Documentation Status PyPI - License

An Apache Airflow provider to interact with MLflow using Operators and Hooks for the following:

  • Registry

  • Deployments

  • Pyfunc

Quick Start

Install and update using pip:

pip install airflow-provider-mlflow

Setting up Connections:

Connection Type: HTTP

  • Local MLflow
  • Hosted with Username/Password
    • Connection Type: HTTP

    • Host: Your MLflow host URL

    • Login: Your MLflow username

    • Password: Your MLflow password

  • Databricks
    • Host: Your Databricks host URL (https://<instance-name>

    • Login: ‘token’

    • Password: Your Databricks token

Examples can be found in the example_dags directory of the repo.


We follow Semantic Versioning for releases. Check CHANGELOG.rst for the latest changes.


Apache License 2.0

Project details

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

airflow-provider-mlflow-1.1.0.tar.gz (14.2 kB view hashes)

Uploaded Source

Built Distribution

airflow_provider_mlflow-1.1.0-py3-none-any.whl (18.2 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page