rhino_health.lib.endpoints.aimodel.aimodel_dataclass#

Module Contents#

Classes#

ModelTypes

str(object='') -> str

AIModel

AIModelRunInput

AIModelTrainInput

class rhino_health.lib.endpoints.aimodel.aimodel_dataclass.ModelTypes#

Bases: str, enum.Enum

str(object=’’) -> str str(bytes_or_buffer[, encoding[, errors]]) -> str

Create a new string object from the given object. If encoding or errors is specified, then the object must expose a data buffer that will be decoded using the given encoding and error handler. Otherwise, returns the result of object.__str__() (if defined) or repr(object). encoding defaults to sys.getdefaultencoding(). errors defaults to ‘strict’.

CLARA_TRAIN = Clara Train#
GENERALIZED_COMPUTE = Generalized Compute#
NVIDIA_FLARE = NVIDIA FLARE#
class rhino_health.lib.endpoints.aimodel.aimodel_dataclass.AIModel(*args, **kwargs)#

Bases: AIModelCreateInput

uid :str#

The unique ID of the AIModel

created_at :str#

When this AIModel was added

name :str#

The name of the AIModel

description :str#

The description of the AIModel

version :int#

The version of the AIModel

base_version_uid :Optional[str] =#

The first version of the AIModel

input_data_schema :str#

The schema pre-run of the input cohorts

output_data_schema :str#

The schema post-run of the output cohorts

project_uid :typing_extensions.Annotated[str, Field(alias='project')]#

The AIModel project

class rhino_health.lib.endpoints.aimodel.aimodel_dataclass.AIModelRunInput(*args, **kwargs)#

Bases: rhino_health.lib.dataclass.RhinoBaseModel

aimodel_uid :str#

The unique ID of the AIModel

input_cohort_uids :List[str]#

A list of the input cohort uids

output_cohort_names_suffix :str#

The suffix given to all output cohorts

run_params :Optional[str] = {}#

The run params code you want to run on the cohorts

timeout_seconds :Optional[int] = 600#

The time before a timeout is declared for the run

sync :Optional[bool] = True#

If True wait for run to end if False let it run in the background

class rhino_health.lib.endpoints.aimodel.aimodel_dataclass.AIModelTrainInput(**data)#

Bases: rhino_health.lib.dataclass.RhinoBaseModel

aimodel_uid :str#

The unique ID of the AIModel

input_cohort_uids :List[str]#

A list of the input cohort uids

validation_cohort_uids :List[str]#

A list of the cohort uids for validation

validation_cohorts_inference_suffix :str#

The suffix given to all output cohorts

config_fed_server :str#

The config for the federated server

config_fed_client :str#

The config for the federated client

timeout_seconds :int#

The time before a timeout is declared for the run