Python Client for MLflow#

status PyPI PyPI License PyPI Python Version ReadTheDocs Build Coverage pre-commit.ci

Python client for MLflow REST API.

Features:

  • Minimal dependencies

  • Unlike MLflow Tracking client all REST API methods and params are exposed to user.

  • MLflow URL is passed via constructor argument instead of env variable, so multiple client instances could be created in the same Python interpreter.

  • Basic and Bearer auth are supported (via constructor args too).

  • All class fields are validated with pydantic.

  • All methods and classes are documented.

Limitations:

  • There is no integration with ML frameworks and libraries. You should use official MLflow client instead.

  • There is no integration with S3 or other artifact storage type. You should access it directly with boto3 or other client.

  • Supported MLflow versions: from 1.17.0 to 1.23.0. It is possible to use client with older MLflow versions (e.g. 1.10.0), but this is not guaranteed.

  • Only Python 3.7+ is supported. Python 3.6 and lower already reached end of life.