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.0to1.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.
Documentation
See https://mlflow-rest-client.readthedocs.io/
Contribution guide
See CONTRIBUTING.rst
Security
See SECURITY.rst
Installation
Stable release
Stable version is released on every tag to master branch. Please use stable releases on production environment. Version example: 2.0.0
pip install mlflow-rest-client==2.0.0 # exact version
pip install mlflow-rest-client # latest release
Development release
Development version is released on every commit to dev branch. You can use them to test some new features before official release. Version example: 2.0.0.dev5
pip install mlflow-rest-client==2.0.0.dev5 # exact dev version
pip install --pre mlflow-rest-client # latest dev version
Development
Clone repo:
git clone [email protected]:MobileTeleSystems/mlflow-rest-client.git
cd mlflow-rest-client
Install dependencies for development:
pip install -r requirements-dev.txt
Install pre-commit hooks:
pre-commit install
pre-commit autoupdate
pre-commit install-hooks
Test pre-commit hooks run:
pre-commit run --all-files -v
Usage
Make sure you have an MLflow Tracking Server running.
from mlflow_rest_client import MLflowRESTClient
client = MLflowRESTClient("https://mlflow.domain", ignore_ssl_check=True)
experiment = client.get_or_create_experiment("experiment_name")
run = client.create_run(experiment.id)
See sample.py for more examples.
![[pre-commit.ci] pre-commit autoupdate](https://avatars.githubusercontent.com/in/68672?v=4)
1 Oct 31, 2021
2.6k Dec 23, 2022
2 Nov 15, 2021
6 Dec 27, 2022
16.7k Dec 28, 2022
1 May 03, 2022
13 Sep 24, 2022
8 May 27, 2022
9k Jan 03, 2023
2.2k Jan 01, 2023
1.3k Dec 28, 2022
1.1k Dec 28, 2022
53 Dec 12, 2022
19 Aug 30, 2022
106 Oct 14, 2022
223 Dec 07, 2022
7 May 30, 2022
1k Dec 27, 2022
5.6k Dec 27, 2022
1 Dec 15, 2021