Python Client for MLflow#
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
to1.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.