Source code for src.v0.models.utility

from pydantic import AliasChoices, Field, constr

from ... import DOTModel
from .meta import VertexMetaData, VertexMetaDataResponse



[docs]
class DiscreteUtilityData(DOTModel):
    parents_uuid: list[str]
    values: list[list[float]]




[docs]
class ContinuousUtilityData(DOTModel):
    parents_uuid: list[str]




[docs]
class UtilityData(DOTModel):
    dtype: DiscreteUtilityData | ContinuousUtilityData | None = None
    unit: str | None = None




[docs]
class Utility(VertexMetaData, UtilityData):
    pass




[docs]
class UtilityResponse(VertexMetaDataResponse, Utility):
    id: str = Field(validation_alias=AliasChoices("T.id", "id"))
    label: constr(to_lower=True) = Field(
        validation_alias=AliasChoices("T.label", "label")
    )