Fix errors

JuliaReinforcementLearning · Mar 26, 2024 · f01afec · f01afec
1 parent 9640fe6
commit f01afec
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/src/ReinforcementLearningEnvironments/src/environments/examples/TicTacToeEnv.jl b/src/ReinforcementLearningEnvironments/src/environments/examples/TicTacToeEnv.jl
@@ -66,8 +66,8 @@ end
 RLBase.players(::TicTacToeEnv) = (Player(:Cross), Player(:Nought))
 
 RLBase.state(env::TicTacToeEnv) = state(env, Observation{Int}(), Player(:Any))
-RLBase.state(env::TicTacToeEnv, ::Observation{BitArray{3}}, p) = env.board
-RLBase.state(env::TicTacToeEnv, ::RLBase.AbstractStateStyle) = state(env::TicTacToeEnv, Observation{Int}(), 1)
+RLBase.state(env::TicTacToeEnv, ::Observation{BitArray{3}}, player) = env.board
+RLBase.state(env::TicTacToeEnv, ::RLBase.AbstractStateStyle) = state(env::TicTacToeEnv, Observation{Int}(), Player(1))
 RLBase.state(env::TicTacToeEnv, ::Observation{Int}, player::Player) =
     get_tic_tac_toe_state_info()[env].index
 

diff --git a/src/ReinforcementLearningEnvironments/test/environments/wrappers/wrappers.jl b/src/ReinforcementLearningEnvironments/test/environments/wrappers/wrappers.jl
@@ -149,6 +149,10 @@
 end
 
 @testset "Wrapped Env run" begin
+    Base.@kwdef mutable struct LotteryEnv <: AbstractEnv
+        reward::Union{Nothing, Int} = nothing
+    end
+
     struct LotteryAction{a}
         function LotteryAction(a)
             new{a}()