diff --git a/expo/experimenter/mcts.py b/expo/experimenter/mcts.py index e06169a70..f0db72841 100644 --- a/expo/experimenter/mcts.py +++ b/expo/experimenter/mcts.py @@ -22,6 +22,7 @@ class MCTSExperimenter(Experimenter): state=self.state, reflection=self.args.reflection, rollouts=self.args.rollouts, + load_tree=self.args.load_tree, ) best_node = best_nodes["global_best"] dev_best_node = best_nodes["dev_best"]