Module `wallaroo.model_config`

Classes

class ModelConfig (gql_client: gql.client.Client)

Create a model configs client with the specified gql client.

def create_model_config(self, model_id: int, runtime: str, tensor_fields: List[str] = None, filter_threshold: float = None) ‑> dict: Creates a model config for an uploaded model file. The model id and runtime must be specified.
def get_model_config(self, model_config_id: int) ‑> Optional[dict]: Get a model config specified by the id. Returns None if it does not exist.
def get_model_configs(self) ‑> List[dict]: Get a list of the current model configs in the system.