Lections/gym-tic-tac-toe
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

checking the diagonals outside the i loop

Browse Source
This commit is contained in:
tsu-nera 2017-06-13 19:57:32 +09:00
parent d6dbdeea7b
commit a79bcf3241
1 changed files with 7 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
gym_tic_tac_toe/envs/tic_tac_toe_env.py
View File

@ -33,14 +33,17 @@ class TicTacToeEnv(gym.Env):
# check game over # check game over
for i in range(3): for i in range(3):
# horizontals and verticals and cross # horizontals and verticals
if ((board[i * 3] == p and board[i * 3 + 1] == p and board[i * 3 + 2] == p) if ((board[i * 3] == p and board[i * 3 + 1] == p and board[i * 3 + 2] == p)
or (board[i + 0] == p and board[i + 3] == p and board[i + 6] == p) or (board[i + 0] == p and board[i + 3] == p and board[i + 6] == p)):
or (board[0] == p and board[4] == p and board[8] == p)
or (board[2] == p and board[4] == p and board[6] == p)):
reward = p reward = p
done = True done = True
break break
# diagonals
if((board[0] == p and board[4] == p and board[8] == p)
or (board[2] == p and board[4] == p and board[6] == p)):
reward = p
done = True
return self.state, reward, done, {} return self.state, reward, done, {}
def _reset(self): def _reset(self):