Skip to content

Commit 9bfe4cf

Browse files
Small improvements for TicTacToeEnv (#692)
* More explicit logging in TicTacToeEnv * Fix type instability in TicTacToeEnv
1 parent c70f4f0 commit 9bfe4cf

File tree

1 file changed

+4
-4
lines changed
  • src/ReinforcementLearningEnvironments/src/environments/examples

1 file changed

+4
-4
lines changed

src/ReinforcementLearningEnvironments/src/environments/examples/TicTacToeEnv.jl

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -58,9 +58,9 @@ RLBase.legal_action_space(env::TicTacToeEnv, p) =
5858

5959
function RLBase.legal_action_space_mask(env::TicTacToeEnv, p)
6060
if is_win(env, CROSS) || is_win(env, NOUGHT)
61-
zeros(false, 9)
61+
falses(9)
6262
else
63-
vec(view(env.board, :, :, 1))
63+
vec(env.board[:, :, 1])
6464
end
6565
end
6666

@@ -135,7 +135,7 @@ end
135135

136136
function get_tic_tac_toe_state_info()
137137
if isempty(TIC_TAC_TOE_STATE_INFO)
138-
@info "initializing state info..."
138+
@info "initializing tictactoe state info cache..."
139139
t = @elapsed begin
140140
n = 1
141141
root = TicTacToeEnv()
@@ -160,7 +160,7 @@ function get_tic_tac_toe_state_info()
160160
end
161161
end
162162
end
163-
@info "finished initializing state info in $t seconds"
163+
@info "finished initializing tictactoe state info cache in $t seconds"
164164
end
165165
TIC_TAC_TOE_STATE_INFO
166166
end

0 commit comments

Comments
 (0)