airflow-provider-mlflow

Apache Airflow provider for MLflow


License
Apache-2.0
Install
pip install airflow-provider-mlflow==1.1.0

Documentation

MLflow Provider Package for Apache Airflow

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

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
  • 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