(self, outputs: List[CompletionOutput], node: Type[MCTSNode])
| 112 | node.update_recursive(value_estimate, self.root) |
| 113 | |
| 114 | def expand_node(self, outputs: List[CompletionOutput], node: Type[MCTSNode]) -> None: |
| 115 | if self.config.remove_duplicate: |
| 116 | dedup_outputs = [] |
| 117 | dedup_keys = set() |
| 118 | for output in outputs: |
| 119 | key = output.text.strip() |
| 120 | if not key in dedup_keys: |
| 121 | dedup_keys.add(key) |
| 122 | dedup_outputs.append(output) |
| 123 | outputs = dedup_outputs |
| 124 | for idx, output in enumerate(outputs): |
| 125 | prior_prob = np.exp(output.cumulative_logprob / len(output.token_ids)) |
| 126 | step_result, parser_result = self.step_unwrap(output.text.strip()) |
| 127 | self.create_child(step_result, parser_result, node, prior_prob, idx) |
| 128 | |
| 129 | def create_child( |
| 130 | self, |
no test coverage detected