def to_tree_transform(cls, model, ctx): selector = OrderedDict() node = OrderedDict() labels = list(model.selector) values = [] for l in labels: values.append(model.selector[l]) selector['labels'] = labels selector['transforms'] = values node['inputs']= list(model.inputs) node['outputs'] = list(model.outputs) node['selector'] = selector node['label_mapper'] = model.label_mapper node['undefined_transform_value'] = model.undefined_transform_value return yamlutil.custom_tree_to_tagged_tree(node, ctx)
def to_tree_transform(cls, model, ctx): node = OrderedDict() if isinstance(model, LabelMapperArray): node['mapper'] = model.mapper if isinstance(model, (LabelMapperDict, LabelMapperRange)): mapper = OrderedDict() labels = list(model.mapper) transforms = [] for k in labels: transforms.append(model.mapper[k]) if isiterable(labels[0]): labels = [list(l) for l in labels] mapper['labels'] = labels mapper['models'] = transforms node['mapper'] = mapper node['inputs'] = list(model.inputs) if model.inputs_mapping is not None: node['inputs_mapping'] = model.inputs_mapping return yamlutil.custom_tree_to_tagged_tree(node, ctx)