in src/braket/schema_common/schema_base.py [0:0]
def import_schema_module(schema: BraketSchemaBase):
"""
Imports the module that holds the schema given the schema
Args:
schema (BraketSchemaBase): The schema
Returns:
Module of the schema
Raises:
ModuleNotFoundError: If the schema module cannot be found according to
schema header
Examples:
>> schema = BraketSchemaBase.parse_raw(json_string)
>> module = import_schema_module(schema)
>> module.AnnealingTaskResult.parse_raw(json_string)
"""
return schema.braketSchemaHeader.import_schema_module()