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)
