fix getstate

JuliaReinforcementLearning · Sep 5, 2018 · b075d84 · b075d84
1 parent c9626ee
commit b075d84
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ReinforcementLearningEnvironmentGym.jl b/src/ReinforcementLearningEnvironmentGym.jl
@@ -46,7 +46,7 @@ end
 interact!(action, env::GymEnv) = interactgym!(action, env)
 interact!(action::Int64, env::GymEnv) = interactgym!(action - 1, env)
 reset!(env::GymEnv) = env.pyobj[:reset]()
-getstate(env::GymEnv) = (Float64[env.pyobj[:env][:state]...], false) # doesn't work for all envs
+getstate(env::GymEnv) = (env.state[1], false) 
 
 plotenv(env::GymEnv, s, a, r, d) = env.pyobj[:render]()
 """