add task arg for tree visualization

This commit is contained in:
Yizhou Chi 2024-10-18 14:29:36 +08:00
parent 7a38165e6b
commit f97ad720b0

View file

@ -20,4 +20,4 @@ if __name__ == "__main__":
root = mcts.root_node
G = nx.DiGraph()
build_tree_recursive(G, "0", root)
visualize_tree(G, save_path="results/tree.png")
visualize_tree(G, save_path=f"results/{args.task}-tree.png")