def transform_context(self, context, builder):
    context['tree'] = builder.tree
    return context
def transform_context(self, context, builder):
    context['tree'] = builder.tree
    return context