`nemo_rl.data.energon.multimodal.model_families`#

Module Contents#

`supports_model_families`	Declare the model families supported by a cooker or task encoder.
`get_supported_model_families`	Read one component’s immutable model-family declaration.
`supports_model_family`	Return whether a declared component supports one model family.

nemo_rl.data.energon.multimodal.model_families.ALL_MODEL_FAMILIES: Final[nemo_rl.data.energon.multimodal.model_families.AllModelFamilies]#: ‘*’

nemo_rl.data.energon.multimodal.model_families.KNOWN_MODEL_FAMILIES: Final#: ‘cast(…)’

nemo_rl.data.energon.multimodal.model_families._MODEL_FAMILIES_ATTRIBUTE#: ‘nemo_rl_supported_model_families’

nemo_rl.data.energon.multimodal.model_families.supports_model_families( *model_families: nemo_rl.data.energon.multimodal.model_families.SupportedModelFamily, ) → collections.abc.Callable[[nemo_rl.data.energon.multimodal.model_families.ComponentT], nemo_rl.data.energon.multimodal.model_families.ComponentT]#: Declare the model families supported by a cooker or task encoder.

nemo_rl.data.energon.multimodal.model_families.get_supported_model_families( component: object, ) → frozenset[nemo_rl.data.energon.multimodal.model_families.SupportedModelFamily]#: Read one component’s immutable model-family declaration.

nemo_rl.data.energon.multimodal.model_families.supports_model_family( component: object, model_family: nemo_rl.data.energon.multimodal.model_families.ModelFamily, ) → bool#: Return whether a declared component supports one model family.

nemo_rl.data.energon.multimodal.model_families.__all__#: [‘ALL_MODEL_FAMILIES’, ‘KNOWN_MODEL_FAMILIES’, ‘ModelFamily’, ‘SupportedModelFamily’, ‘get_supported…