`nemo_microservices.data_designer.config.models`#

Module Contents#

Classes#

`Distribution`
`DistributionType`
`ImageContext`
`ImageFormat`
`InferenceParameters`
`ManualDistribution`
`ManualDistributionParams`
`Modality`
`ModalityContext`
`ModalityDataType`
`ModelConfig`
`ModelProvider`
`UniformDistribution`
`UniformDistributionParams`

Functions#

load_model_configs

Data#

`DistributionParamsT`
`DistributionT`
`logger`

API#

class nemo_microservices.data_designer.config.models.Distribution(/, **data: typing.Any)#

Bases: abc.ABC, nemo_microservices.data_designer.config.base.ConfigBase, typing.Generic[nemo_microservices.data_designer.config.models.DistributionParamsT]

distribution_type: nemo_microservices.data_designer.config.models.DistributionType#: None

params: nemo_microservices.data_designer.config.models.DistributionParamsT#: None

abstract sample() → float#

nemo_microservices.data_designer.config.models.DistributionParamsT#: ‘TypeVar(…)’

nemo_microservices.data_designer.config.models.DistributionT: typing_extensions.TypeAlias#: None

class nemo_microservices.data_designer.config.models.DistributionType#

Bases: str, enum.Enum

MANUAL#: ‘manual’

UNIFORM#: ‘uniform’

class nemo_microservices.data_designer.config.models.ImageContext(/, **data: typing.Any)#

Bases: nemo_microservices.data_designer.config.models.ModalityContext

get_context(record: dict) → dict[str, Any]#

image_format: Optional[nemo_microservices.data_designer.config.models.ImageFormat]#: None

modality: nemo_microservices.data_designer.config.models.Modality#: None

class nemo_microservices.data_designer.config.models.ImageFormat#

Bases: str, enum.Enum

GIF#: ‘gif’

JPEG#: ‘jpeg’

JPG#: ‘jpg’

PNG#: ‘png’

WEBP#: ‘webp’

class nemo_microservices.data_designer.config.models.InferenceParameters(/, **data: typing.Any)#

Bases: nemo_microservices.data_designer.config.base.ConfigBase

extra_body: Optional[dict[str, Any]]#: None

property generate_kwargs: dict[str, Union[float, int]]#

max_parallel_requests: int#: ‘Field(…)’

max_tokens: Optional[int]#: ‘Field(…)’

temperature: Optional[Union[float, nemo_microservices.data_designer.config.models.DistributionT]]#: None

timeout: Optional[int]#: ‘Field(…)’

top_p: Optional[Union[float, nemo_microservices.data_designer.config.models.DistributionT]]#: None

class nemo_microservices.data_designer.config.models.ManualDistribution#

Bases: nemo_microservices.data_designer.config.models.Distribution[nemo_microservices.data_designer.config.models.ManualDistributionParams]

distribution_type: Optional[nemo_microservices.data_designer.config.models.DistributionType]#: ‘manual’

params: nemo_microservices.data_designer.config.models.ManualDistributionParams#: None

sample() → float#

class nemo_microservices.data_designer.config.models.ManualDistributionParams(/, **data: typing.Any)#

Bases: nemo_microservices.data_designer.config.base.ConfigBase

values: List[float]#: ‘Field(…)’

weights: Optional[List[float]]#: None

class nemo_microservices.data_designer.config.models.Modality#

Bases: str, enum.Enum

IMAGE#: ‘image’

class nemo_microservices.data_designer.config.models.ModalityContext(/, **data: typing.Any)#

Bases: abc.ABC, pydantic.BaseModel

column_name: str#: None

data_type: nemo_microservices.data_designer.config.models.ModalityDataType#: None

abstract get_context(record: dict) → dict[str, Any]#

modality: nemo_microservices.data_designer.config.models.Modality#: None

class nemo_microservices.data_designer.config.models.ModalityDataType#

Bases: str, enum.Enum

BASE64#: ‘base64’

URL#: ‘url’

class nemo_microservices.data_designer.config.models.ModelConfig(/, **data: typing.Any)#

Bases: nemo_microservices.data_designer.config.base.ConfigBase

alias: str#: None

inference_parameters: nemo_microservices.data_designer.config.models.InferenceParameters#: ‘Field(…)’

model: str#: None

provider: Optional[str]#: None

class nemo_microservices.data_designer.config.models.ModelProvider(/, **data: typing.Any)#

Bases: nemo_microservices.data_designer.config.base.ConfigBase

api_key: Optional[str]#: None

endpoint: str#: None

extra_body: Optional[dict[str, Any]]#: None

name: str#: None

provider_type: str#: ‘openai’

class nemo_microservices.data_designer.config.models.UniformDistribution#

Bases: nemo_microservices.data_designer.config.models.Distribution[nemo_microservices.data_designer.config.models.UniformDistributionParams]

distribution_type: Optional[nemo_microservices.data_designer.config.models.DistributionType]#: ‘uniform’

params: nemo_microservices.data_designer.config.models.UniformDistributionParams#: None

sample() → float#

class nemo_microservices.data_designer.config.models.UniformDistributionParams(/, **data: typing.Any)#

Bases: nemo_microservices.data_designer.config.base.ConfigBase

high: float#: None

low: float#: None

nemo_microservices.data_designer.config.models.load_model_configs( model_configs: Union[list[nemo_microservices.data_designer.config.models.ModelConfig], str, pathlib.Path], ) → list[nemo_microservices.data_designer.config.models.ModelConfig]#

nemo_microservices.data_designer.config.models.logger#: ‘getLogger(…)’

nemo_microservices.data_designer.config.models#

Module Contents#

Classes#

Functions#

Data#

API#

`nemo_microservices.data_designer.config.models`#