from src.tree import gen_tree if __name__ == '__main__': gen_tree('data')