Skip to content

Commit c4601a1

Browse files
authored
fix: fallback to AlphaZero if RPO fails (#25)
1 parent 5912381 commit c4601a1

File tree

1 file changed

+8
-1
lines changed

1 file changed

+8
-1
lines changed

src/model/policy.cpp

Lines changed: 8 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -45,7 +45,14 @@ void Policy::get_policy(const std::vector<double> &q_values, const std::vector<d
4545
alpha_zero(q_values, pi_values, counts, result);
4646
break;
4747
case RPO:
48-
mcts_rpo(q_values, pi_values, counts, result);
48+
try {
49+
mcts_rpo(q_values, pi_values, counts, result);
50+
}
51+
catch (std::runtime_error &e) {
52+
result.clear();
53+
alpha_zero(q_values, pi_values, counts, result);
54+
}
55+
4956
break;
5057
default:
5158
throw std::invalid_argument("Invalid policy type");

0 commit comments

Comments
 (0)