diff --git a/expo/experimenter/mcts.py b/expo/experimenter/mcts.py index bd803bff1..fa42cb070 100644 --- a/expo/experimenter/mcts.py +++ b/expo/experimenter/mcts.py @@ -31,9 +31,6 @@ class MCTSExperimenter(Experimenter): dev_best_node = best_nodes["dev_best"] score_dict = best_nodes["scores"] - self.copy_notebook(best_node, "best") - self.copy_notebook(dev_best_node, "dev_best") - text, num_generated_codes = get_tree_text(mcts.root_node) text += f"Generated {num_generated_codes} unique codes.\n" text += f"Best node: {best_node.id}, score: {best_node.raw_reward}\n" @@ -55,6 +52,8 @@ class MCTSExperimenter(Experimenter): } ] self.save_result(results) + self.copy_notebook(best_node, "best") + self.copy_notebook(dev_best_node, "dev_best") def copy_notebook(self, node, name): node_dir = node.get_node_dir()