Fix129 (#130)

* fix 129

* add tests

* fix tests

Author: findmyway

src/environments/wrappers/DefaultStateStyle.jl

@@ -12,3 +12,10 @@ Reset the result of `DefaultStateStyle` without changing the original behavior.
 DefaultStateStyleEnv{S}(env::E) where {S,E} = DefaultStateStyleEnv{S,E}(env)
 
 RLBase.DefaultStateStyle(::DefaultStateStyleEnv{S}) where {S} = S
+
+RLBase.state(env::DefaultStateStyleEnv{S}) where S = state(env.env, S)
+RLBase.state(env::DefaultStateStyleEnv, ss::RLBase.AbstractStateStyle) = state(env.env, ss)
+RLBase.state(env::DefaultStateStyleEnv{S}, player) where S = state(env.env, S, player)
+
+RLBase.state_space(env::DefaultStateStyleEnv{S}) where S = state_space(env.env, S)
+RLBase.state_space(env::DefaultStateStyleEnv, ss::RLBase.AbstractStateStyle) = state_space(env.env, ss)

test/environments/wrappers/wrappers.jl

@@ -21,6 +21,16 @@
 
         RLBase.test_interfaces!(env′)
         RLBase.test_runnable!(env′)
+
+        # fix #129
+        env = TicTacToeEnv()
+        s_env = state(env)
+        @test s_env ∈ state_space(env)
+
+        E = DefaultStateStyleEnv{Observation{Int}()}(env)  # error probably somewhere here
+        s = state(E)
+        @test s isa Int
+        @test s ∈ state_space(E)
     end
 
     @testset "MaxTimeoutEnv" begin