fix tests

Author: findmyway

src/examples/PigEnv.jl

@@ -36,7 +36,7 @@ prob(env::PigEnv, ::ChancePlayer) = fill(1 / 6, 6)  # TODO: uniform distribution
 
 state(env::PigEnv, ::Observation{Vector{Int}}, p) = env.scores
 state_space(env::PigEnv, ::Observation, p) =
-    [0..(PIG_TARGET_SCORE + PIG_N_SIDES - 1) for _ in env.scores]
+    Space([0..(PIG_TARGET_SCORE + PIG_N_SIDES - 1) for _ in env.scores])
 
 is_terminated(env::PigEnv) = any(s >= PIG_TARGET_SCORE for s in env.scores)
 

src/examples/TicTacToeEnv.jl

@@ -75,7 +75,7 @@ current_player(env::TicTacToeEnv) = env.player
 players(env::TicTacToeEnv) = (CROSS, NOUGHT)
 
 state(env::TicTacToeEnv, ::Observation{BitArray{3}}, p) = env.board
-state_space(env::TicTacToeEnv, ::Observation{BitArray{3}}, p) = fill(false..true, 3, 3, 3)
+state_space(env::TicTacToeEnv, ::Observation{BitArray{3}}, p) = Space(fill(false..true, 3, 3, 3))
 state(env::TicTacToeEnv, ::Observation{Int}, p) = get_tic_tac_toe_state_info()[env].index
 state_space(env::TicTacToeEnv, ::Observation{Int}, p) =
     Base.OneTo(length(get_tic_tac_toe_state_info()))