wallaroo.model_registry

class ModelRegistry(wallaroo.object.Object):

An instance of an external Model Registry. Currently, the supported registries are: DataBricks

ModelRegistry(client: wallaroo.client.Client, data: Dict[str, Any])

Base constructor.

Each object requires:

a GraphQL client - in order to fill its missing members dynamically
an initial data blob - typically from unserialized JSON, contains at
least the data for required members (typically the object's primary key) and optionally other data members.

def name(*args, **kwargs):

def id(*args, **kwargs):

def url(*args, **kwargs):

def created_at(*args, **kwargs):

def updated_at(*args, **kwargs):

def workspaces(*args, **kwargs):

def add_registry_to_workspace(self, workspace_id: int):

def remove_registry_from_workspace(self, workspace_id: int):

def list_models(self):

def upload_model( self, name: str, path: str, framework: wallaroo.framework.Framework, input_schema: pyarrow.lib.Schema, output_schema: pyarrow.lib.Schema, requirements: List[str] = []) -> wallaroo.model_version.ModelVersion:

Parameters

name str A descriptive name to set
path str The DBFS path to the artifact file to upload.
framework Framework The ML framework that this model uses (ex. Framework.ONNX).
requirements List[str] An optional list of python requirements needed to run this model.
input_schema pa.Schema A PyArrow schema describing inputs to this model.
output_schema pa.Schema A PyArrow schema describing the outputs of this model.

Returns

Model An instance of the Wallaroo model that is being created.

class ModelRegistriesList(typing.List[wallaroo.model_registry.ModelRegistry]):

Wraps a list of Model Registries for display in a display-aware environment like Jupyter.

Inherited Members

builtins.list: list; clear; copy; append; insert; extend; pop; remove; index; count; reverse; sort

class RegisteredModelVersion(wallaroo.wallaroo_ml_ops_api_client.models.registered_model_version.RegisteredModelVersion):

An MLFlow version of a [RegisteredModel]. Versions start at 1.

Attributes: creation_timestamp (int): Timestamp in milliseconds from epoch current_stage (str): last_updated_timestamp (int): Timestamp in milliseconds from epoch name (str): run_id (str): source (str): status (str): version (str): description (Union[None, Unset, str]): run_link (Union[None, Unset, str]): user_id (Union[None, Unset, str]):

RegisteredModelVersion( client: wallaroo.client.Client, data: wallaroo.wallaroo_ml_ops_api_client.models.registered_model_version.RegisteredModelVersion, registry_id: str)

Method generated by attrs for class RegisteredModelVersion.

updated_at

def upload(self, framework: Framework, requirements: List[str] = []): # Uploads this Model's source code to your Wallaroo instance.

#:param: framework Framework The ML framework that this model uses (ex. Framework.ONNX).
#:param: requirements List[str] An optional list of python requirements needed to run this model.
#:return: Model An instance of the Wallaroo model that is being created.
#
from .wallaroo_ml_ops_api_client.api.model.upload_from_registry import sync
from .wallaroo_ml_ops_api_client.models.upload_from_registry_request import (
    UploadFromRegistryRequest,
)
from .wallaroo_ml_ops_api_client.models.conversion import Conversion

ret = sync(
    client=self._client.mlops(),
    body=UploadFromRegistryRequest(
        Conversion(
            framework=framework._to_openapi_framework(),
            requirements=requirements,
        ),
        self.name,
        self.registry_id,
        self.version,
        "private",
        self._client.get_current_workspace()._id,
    ),
)

if ret is None:
    raise Exception("Failed to start upload.")

return Model(client=self._client, data={"id": ret.model_id})

def list_artifacts(self):

Inherited Members

wallaroo.wallaroo_ml_ops_api_client.models.registered_model_version.RegisteredModelVersion: creation_timestamp; current_stage; last_updated_timestamp; name; run_id; source; status; version; description; run_link; user_id; additional_properties; to_dict; from_dict; additional_keys

class DbfsFileList(typing.List[wallaroo.wallaroo_ml_ops_api_client.models.dbfs_list_response_file_with_full_path.DbfsListResponseFileWithFullPath]):

Built-in mutable sequence.

If no argument is given, the constructor creates a new empty list. The argument must be an iterable if specified.

Inherited Members

builtins.list: list; clear; copy; append; insert; extend; pop; remove; index; count; reverse; sort

class RegisteredModelVersionsList(typing.List[wallaroo.model_registry.RegisteredModelVersion]):

Wraps a list of Model Registries for display in a display-aware environment like Jupyter.

Inherited Members

builtins.list: list; clear; copy; append; insert; extend; pop; remove; index; count; reverse; sort

class RegisteredModel:

This is a reference to a Registered Model in an external registry.

RegisteredModel( client: wallaroo.client.Client, data: wallaroo.wallaroo_ml_ops_api_client.models.registered_model.RegisteredModel, registry_id: str)

def versions(self):

def list_artifacts(self):

class RegisteredModels(typing.List[wallaroo.model_registry.RegisteredModel]):

This is a list of Registered Models in an external registry.

Inherited Members

builtins.list: list; clear; copy; append; insert; extend; pop; remove; index; count; reverse; sort