mlflow-aws-rds-iam 0.3.0

MLflow plugins adding AWS RDS IAM auth for tracking and model registry stores

MLflow AWS RDS IAM Auth Plugins

MLflow plugins adding AWS RDS IAM authentication for tracking store and model registry store.

Requirements

• Python 3.10+
• MLflow 2+
• Boto3

Only Postgres and MySQL are supported (the AWS RDS database engines supported by MLflow as remote backend store).

Installation

pip install mlflow-aws-rds-iam

Getting Started

The plugins are mainly intented for the use of MLflow in server mode, allowing AWS RDS IAM authentication for the backend store.

mlflow server --backend-store-uri <database URI> --artifacts-destination <artifact store URI>

Configuration

If a password is provided in the database URI, AWS IAM authentication will be skipped, effectively disabling the plugins.

Credentials are required in order to use AWS IAM authentication. Refer to Boto3 documentation for configuration instructions.

SSL connection to the database is enabled by default, but can be disabled with the environment variable MLFLOW_DISABLE_DB_SSL=true.

Debug log of the plugins and underlying libraries (boto3, urllib3) can be enabled with MLFLOW_AWS_RDS_IAM_DEBUG=true.

License

This project is licensed under the terms of the MIT license.

A yzr Free and Open Source project.