Actually return the generated moves

2017-04-04 17:45:58 +02:00 · 2017-04-04 17:45:58 +02:00 · 8bed7b4aae
commit 8bed7b4aae
parent 74045a7615
1 changed files with 2 additions and 0 deletions
--- a/gym_tic_tac_toe/envs/tic_tac_toe_env.py
+++ b/gym_tic_tac_toe/envs/tic_tac_toe_env.py
@ -55,8 +55,10 @@ class TicTacToeEnv(gym.Env):
    def move_generator(self):
        moves = []
        for i in range (9):
+            
            if (self.state['board'][i]== 0):
                p = self.state['on_move']
                m = [p, i]
                moves.append(m)
+        return moves