diff --git a/expo/experimenter/mcts.py b/expo/experimenter/mcts.py index fa42cb070..22b480caf 100644 --- a/expo/experimenter/mcts.py +++ b/expo/experimenter/mcts.py @@ -36,8 +36,6 @@ class MCTSExperimenter(Experimenter): text += f"Best node: {best_node.id}, score: {best_node.raw_reward}\n" text += f"Dev best node: {dev_best_node.id}, score: {dev_best_node.raw_reward}\n" print(text) - self.save_tree(text) - results = [ { "best_node": best_node.id, @@ -54,6 +52,7 @@ class MCTSExperimenter(Experimenter): self.save_result(results) self.copy_notebook(best_node, "best") self.copy_notebook(dev_best_node, "dev_best") + self.save_tree(text) def copy_notebook(self, node, name): node_dir = node.get_node_dir()