Skip to main content

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


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

sagemaker_mlflow-0.1.0.tar.gz (12.8 kB view details)

Uploaded Source

Built Distribution

sagemaker_mlflow-0.1.0-py3-none-any.whl (24.7 kB view details)

Uploaded Python 3

File details

Details for the file sagemaker_mlflow-0.1.0.tar.gz.

File metadata

  • Download URL: sagemaker_mlflow-0.1.0.tar.gz
  • Upload date:
  • Size: 12.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.0 CPython/3.11.9

File hashes

Hashes for sagemaker_mlflow-0.1.0.tar.gz
Algorithm Hash digest
SHA256 1fe8f7f010f7c68b6b0b46c032cf6a414f20adfc26cbc6a731d3a91b32b9b84f
MD5 49877d1171e8bc6c238b79263849078a
BLAKE2b-256 2e74b8683edc86515e587a173eefc376a58bbae4768a4e4c9c83ba997f509d65

See more details on using hashes here.

File details

Details for the file sagemaker_mlflow-0.1.0-py3-none-any.whl.

File metadata

File hashes

Hashes for sagemaker_mlflow-0.1.0-py3-none-any.whl
Algorithm Hash digest
SHA256 b0dc955e2898de2070b489e982372edafc0ec708634a2e69c21e2570d7308b0c
MD5 27fe21b7a258a67f93f1782024203072
BLAKE2b-256 5e54e2dd3cf3e289e480600649b1e3f573ca54c8df1db1ac78cc55a24279b924

See more details on using hashes here.

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page