AWS Plugin for MLFlow with SageMaker
Project description
SageMaker MLFlow Plugin
What does this Plugin do?
This plugin generates Signature V4 headers in each outgoing request to the SageMaker MLFlow service and also determines the URL of the SageMaker MLFlow service. It generates a token with the SigV4 Algorithm that the service will use to conduct Authentication and Authorization.
Installation
To install this plugin, run the following command inside the directory:
pip install .
Eventually when the plugin gets distributed, it will be installed with:
pip install sagemaker-mlflow
Running this will install the Auth Plugin and mlflow.
To install a specific mlflow version
pip install .
pip install mlflow==2.13
Development details
setup.py
setup.py
Contains the primary entry points for the sdk.
install_requires
Installs mlflow.
entry_points
Contains the entry points for the sdk. See https://mlflow.org/docs/latest/plugins.html#defining-a-plugin
for more details.
Running tests
Setup
To run tests using tox, run:
pip install tox
Installing tox will enable users to run multi-environment tests. On the other hand, if running individual tests in a single environment, feel free to continue to use pytest instead.
Running format checks
tox -e flake8,black-check,typing,twine
Formatting code to comply with format checks
tox -e black-format
Running unit tests
tox --skip-env "black.*|flake8|typing|twine" -- test/unit
Running integration tests
tox --skip-env "black.*|flake8|typing|twine" -- test/integration
Available test environments by default
tox.ini contains support for py39, py310, py311, with mlflow 2.11.* and 2.12.*.
To add test environments on tox for additional versions of python or mlflow, modify the
environment configs in envlist
, as well as deps
and depends
in [testenv]
.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
Built Distribution
Hashes for sagemaker_mlflow-0.1.0-py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | b0dc955e2898de2070b489e982372edafc0ec708634a2e69c21e2570d7308b0c |
|
MD5 | 27fe21b7a258a67f93f1782024203072 |
|
BLAKE2b-256 | 5e54e2dd3cf3e289e480600649b1e3f573ca54c8df1db1ac78cc55a24279b924 |