Python Client for MLflowΒΆ

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

Python client for MLflow REST API.

Features:

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

  • All class fields are validated with pydantic.

  • Basic and Bearer auth is supported.

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

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