fix tests

findmyway · findmyway · commit 9c4cae51f941 · 2022-05-26T22:05:18.000+08:00
diff --git a/src/ReinforcementLearningTrajectories.jl b/src/ReinforcementLearningTrajectories.jl
@@ -1,5 +1,8 @@
 module ReinforcementLearningTrajectories
 
+const RLTrajectories = ReinforcementLearningTrajectories
+export RLTrajectories
+
 include("patch.jl")
 
 include("traces.jl")
diff --git a/test/samplers.jl b/test/samplers.jl
@@ -6,7 +6,7 @@
         action=rand(1:4, 5),
     )
 
-    b = Trajectories.sample(s, t)
+    b = ReinforcementLearningTrajectories.sample(s, t)
 
     @test keys(b) == (:state, :action)
     @test size(b.state) == (3, 4, sz)

Original file line number	Diff line number	Diff line change
`@@ -6,7 +6,7 @@`
`6`	`6`	`action=rand(1:4, 5),`
`7`	`7`	`)`
`8`	`8`
`9`		`- b = Trajectories.sample(s, t)`
	`9`	`+ b = ReinforcementLearningTrajectories.sample(s, t)`
`10`	`10`
`11`	`11`	`@test keys(b) == (:state, :action)`
`12`	`12`	`@test size(b.state) == (3, 4, sz)`