in causalPartition.py [0:0]
def plot_tree(self, result=None):
if 1 in result:
print('treated')
self._plot_tree(result[1], 1, '')
print('non-treated')
self._plot_tree(result[0], 1, '')
else:
self._plot_tree(result, 1, '')