nemo_rl.data.energon.multimodal.registry

Module Contents

Classes

LazyRegistryEntry	Import path and stable implementation version for one registry key.
LazyRegistry	Resolve configured components inside the process that owns the loader.

Functions

selected_registry_identity

Return stable identity data for all configured multimodal components.

Data

RegistryKind
COOKER_REGISTRY
TASK_ENCODER_REGISTRY
__all__

API

nemo_rl.data.energon.multimodal.registry.RegistryKind

None

class nemo_rl.data.energon.multimodal.registry.LazyRegistryEntry

Import path and stable implementation version for one registry key.

import_path: str

None

version: str

None

class nemo_rl.data.energon.multimodal.registry.LazyRegistry(

kind: nemo_rl.data.energon.multimodal.registry.RegistryKind,

)

Resolve configured components inside the process that owns the loader.

Initialization

register(key: str, *, import_path: str, version: str) → None

Register one unused key without importing its implementation.

resolve(key: str) → Any

Import and type-check one configured implementation.

identity(key: str) → dict[str, str]

Return stable fingerprint data without importing the implementation.

resolve_for_model_family(

key: str,

*,

model_family: nemo_rl.data.energon.multimodal.model_families.ModelFamily,

) → Any

Resolve a cooker or task encoder and validate its model family.

_validate(key: str, resolved: Any) → None

nemo_rl.data.energon.multimodal.registry.COOKER_REGISTRY

‘LazyRegistry(…)’

nemo_rl.data.energon.multimodal.registry.TASK_ENCODER_REGISTRY

‘LazyRegistry(…)’

nemo_rl.data.energon.multimodal.registry.selected_registry_identity(

*,

task_encoder: str,

cookers: list[str],

) → dict[str, Any]

Return stable identity data for all configured multimodal components.

nemo_rl.data.energon.multimodal.registry.__all__

[‘COOKER_REGISTRY’, ‘LazyRegistry’, ‘LazyRegistryEntry’, ‘TASK_ENCODER_REGISTRY’, ‘selected_registry…