Adapt TicTacToe fix

JuliaReinforcementLearning · Sep 3, 2024 · b8572cc · b8572cc
1 parent 43a13a4
commit b8572cc
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/src/ReinforcementLearningEnvironments/src/environments/examples/TicTacToeEnv.jl b/src/ReinforcementLearningEnvironments/src/environments/examples/TicTacToeEnv.jl
@@ -70,10 +70,9 @@ end
 
 RLBase.players(::TicTacToeEnv) = (Player(:Cross), Player(:Nought))
 
-RLBase.state(env::TicTacToeEnv, ::Observation, ::DefaultPlayer) = state(env, Observation{Int}(), Player(:Any))
+RLBase.state(env::TicTacToeEnv, o::Observation, ::DefaultPlayer) = state(env, o, Player(:Any))
 RLBase.state(env::TicTacToeEnv, ::Observation{BitArray{3}}, player) = env.board
-RLBase.state(env::TicTacToeEnv, ::Observation{BitArray{3}}) = env.board
-RLBase.state(env::TicTacToeEnv, ::RLBase.AbstractStateStyle) = state(env::TicTacToeEnv, Observation{Int}(), Player(1))
+RLBase.state(env::TicTacToeEnv, ::RLBase.AbstractStateStyle) = state(env::TicTacToeEnv, Observation{Int}(), Player(:Any))
 RLBase.state(env::TicTacToeEnv, ::Observation{Int}, player::Player) =
     get_tic_tac_toe_state_info()[env].index