import state function (#46)

close #45
JuliaML · May 8, 2020 · 7d47ebf · 7d47ebf
1 parent 7987cde
commit 7d47ebf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/envs/cartpole.jl b/src/envs/cartpole.jl
@@ -9,7 +9,7 @@ using LearnBase: DiscreteSet
 using RecipesBase
 using Random: seed!
 
-import Reinforce: reset!, actions, finished, step!
+import Reinforce: reset!, actions, finished, step!, state
 
 export
  CartPole,
@@ -101,4 +101,4 @@ end
  end
 end
 
-end
+end