model-registry 0.1.0

Client for Red Hat OpenDataHub Model Registry

Model Registry Python Client

This library provides a high level interface for interacting with a model registry server.

Basic usage

from model_registry import ModelRegistry

registry = ModelRegistry(server_address="server-address", port=9090, author="author")

model = registry.register_model("my-model",
                                "s3://path/to/model",
                                model_format_name="onnx",
                                model_format_version="1",
                                storage_key="aws-connection-path",
                                storage_path="to/model",
                                version="v2.0",
                                description="lorem ipsum",
                                )

model = registry.get_registered_model("my-model")

version = registry.get_model_version("my-model", "v2.0")

experiment = registry.get_model_artifact("my-model", "v2.0")

Development

Common tasks, such as building documentation and running tests, can be executed using nox sessions.

Use nox -l to list sessions and execute them using nox -s [session].