From d6dbdeea7bda25c454da7093e64d15df8e9b48fb Mon Sep 17 00:00:00 2001 From: tsu-nera Date: Tue, 13 Jun 2017 19:04:03 +0900 Subject: [PATCH] add cross condition --- gym_tic_tac_toe/envs/tic_tac_toe_env.py | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/gym_tic_tac_toe/envs/tic_tac_toe_env.py b/gym_tic_tac_toe/envs/tic_tac_toe_env.py index 71007b3..8ebaabf 100644 --- a/gym_tic_tac_toe/envs/tic_tac_toe_env.py +++ b/gym_tic_tac_toe/envs/tic_tac_toe_env.py @@ -33,9 +33,11 @@ class TicTacToeEnv(gym.Env): # check game over for i in range(3): - # horizontals and verticals - 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)): + # horizontals and verticals and cross + 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[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 break