From f97ad720b0ed9b78921cf57662ca7dbf9430ebb8 Mon Sep 17 00:00:00 2001 From: Yizhou Chi Date: Fri, 18 Oct 2024 14:29:36 +0800 Subject: [PATCH] add task arg for tree visualization --- expo/scripts/visualize_experiment.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/expo/scripts/visualize_experiment.py b/expo/scripts/visualize_experiment.py index 42b4490ec..e2443d0fd 100644 --- a/expo/scripts/visualize_experiment.py +++ b/expo/scripts/visualize_experiment.py @@ -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")