github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks 

for more information, see https://pre-commit.ci
pull/6119/head
pre-commit-ci[bot] 2024-11-08 12:38:37 +00:00
parent 6c619c9992
commit 308960534f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
applications/ColossalChat/coati/reasoner/guided_search/mcts.py
View File

@ -120,14 +120,14 @@ class MCTS(BaseModel):
self.back_propagation(child)
return self.get_best_answer()
def _iter_nodes(self):
to_visit = deque([self.root])
while to_visit:
current_node = to_visit.popleft()
yield current_node
to_visit.extend(current_node.children)
def get_best_answer(self):
best_node = max(self._iter_nodes(), key=lambda node: node.Q, default=self.root)
return best_node.answer