MLflow Provider Package for Apache Airflow

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

• Registry
• Deployments
• Pyfunc

https://mlflow.org/docs/latest/index.html

Quick Start

Install and update using pip:

pip install airflow-provider-mlflow

Setting up Connections:

Connection Type: HTTP

• Local MLflow

  • Host: http://localhost (if running Airflwo in docker: http://host.docker.internal)
  • Port: 5000

• 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>.cloud.databricks.com)
  • Login: 'token'
  • Password: Your Databricks token

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

Changelog

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

License

Apache License 2.0