def assert_inner_types(parent_type, *dagster_types): config_type = resolve_to_config_type(parent_type) config_schema_snapshot = config_schema_snapshot_from_config_type( config_type) all_type_keys = get_recursive_type_keys(snap_from_config_type(config_type), config_schema_snapshot) assert set(all_type_keys) == set( map(lambda x: x.key, map(resolve_to_config_type, dagster_types)))
def build_config_schema_snapshot(pipeline_def): check.inst_param(pipeline_def, 'pipeline_def', PipelineDefinition) all_config_snaps_by_key = {} for mode in pipeline_def.available_modes: run_config_schema = pipeline_def.get_run_config_schema(mode) all_config_snaps_by_key = merge_dicts( all_config_snaps_by_key, {ct.key: snap_from_config_type(ct) for ct in run_config_schema.all_config_types()}, ) return ConfigSchemaSnapshot(all_config_snaps_by_key)