From ced63af18fcc790c4b1bb1548b5494bd2073f9a2 Mon Sep 17 00:00:00 2001 From: Wenbo Hu Date: Thu, 21 Dec 2017 19:31:51 +0800 Subject: [PATCH] fixing bug pass parameterg --- tianshou/core/mcts/mcts.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tianshou/core/mcts/mcts.py b/tianshou/core/mcts/mcts.py index 7edac97..8bb5f06 100644 --- 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")