Module wallaroo.model
Classes
class Model (rest_api_host: str, gql_client: gql.client.Client)-
This class is used to upload models to the model server.
Methods
def get_model(self, model_id: str, version: str) ‑> List[dict]-
"Gets a specific model by model_id/name and version.
def get_model_by_id(self, row_id: int) ‑> Optional[dict]-
Gets a model by its row id.
def get_models(self) ‑> List[dict]-
Get a list of models from the database.
def get_models_by_name(self, model_id: str) ‑> List[dict]-
Get models by model_id/name. Useful for finding all versions of a model.
def upload_model(self, id: str, version: str, path: str) ‑> Optional[dict]-
Upload model through rest server. Returns the database record for the model.