Actually return the generated moves

This commit is contained in:
Nicolai Czempin 2017-04-04 17:45:58 +02:00
parent 74045a7615
commit 8bed7b4aae

View File

@ -55,8 +55,10 @@ class TicTacToeEnv(gym.Env):
def move_generator(self): def move_generator(self):
moves = [] moves = []
for i in range (9): for i in range (9):
if (self.state['board'][i]== 0): if (self.state['board'][i]== 0):
p = self.state['on_move'] p = self.state['on_move']
m = [p, i] m = [p, i]
moves.append(m) moves.append(m)
return moves