fixing bug pass parameterg

2017-12-21 19:31:51 +08:00 · 2017-12-21 19:31:51 +08:00 · ced63af18f
commit ced63af18f
parent 00d2aa86bf
1 changed files with 1 additions and 1 deletions
--- a/tianshou/core/mcts/mcts.py
+++ b/tianshou/core/mcts/mcts.py
@ -73,7 +73,7 @@ class UCTNode(MCTSNode):
    def valid_mask(self, simulator):
        # let all invalid actions be illeagel in mcts
        if self.mask is None:
-            self.mask = simulator.simulate_is_valid_list(self.state, range(self.action_num - 1))
+            self.mask = simulator.simulate_is_valid_list(self.state, range(self.action_num))
        self.ucb[self.mask] = -float("Inf")