fix visualization bug

This commit is contained in:
Yizhou Chi 2024-10-17 10:34:05 +08:00
parent 1d22466ac5
commit 6646983a25

View file

@ -19,5 +19,5 @@ if __name__ == "__main__":
mcts.load_tree()
root = mcts.root_node
G = nx.DiGraph()
tree = build_tree_recursive(G, "0", root)
visualize_tree(tree, save_path="results/tree.png")
build_tree_recursive(G, "0", root)
visualize_tree(G, save_path="results/tree.png")