krugd/rustic_mcts
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fixed missing non terminal error

Browse Source
This commit is contained in:
David Kruger 2025-06-30 20:08:18 -07:00
parent a2711f341f
commit a7102a0e44
2 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
Cargo.toml
View File

@ -16,7 +16,7 @@ rand = "~0.9"
thiserror = "~2.0" thiserror = "~2.0"
[dev-dependencies] [dev-dependencies]
divan = "0.1.21" divan = "~0.1"
[[bench]] [[bench]]
name = "e2e" name = "e2e"

9
src/mcts.rs
View File

@ -75,9 +75,16 @@ impl<'conf, S: GameState + std::fmt::Debug> MCTS<'conf, S> {
if !selected_node.state.is_terminal() { if !selected_node.state.is_terminal() {
self.expand(selected_id); self.expand(selected_id);
let children: &Vec<usize> = &self.arena.get_node(selected_id).children; let children: &Vec<usize> = &self.arena.get_node(selected_id).children;
let random_child: usize = *children.choose(&mut rand::rng()).unwrap(); match children.choose(&mut rand::rng()) {
Some(&random_child) => {
selected_id = random_child; selected_id = random_child;
} }
None => {
// We ran out of nodes
return Err(MCTSError::NonTerminalGame);
}
}
}
let rewards = self.simulate(selected_id); let rewards = self.simulate(selected_id);
self.backprop(selected_id, &rewards); self.backprop(selected_id, &rewards);