fix tests

findmyway · findmyway · commit 85f531d92760 · 2022-06-12T13:00:48.000+08:00
diff --git a/src/samplers.jl b/src/samplers.jl
@@ -99,7 +99,7 @@ mutable struct NStepBatchSampler{traces}
     rng::Any
 end
 
-NStepBatchSampler(; kw...) = NStepBatchSampler{SSART}(; kw...)
+NStepBatchSampler(; kw...) = NStepBatchSampler{SS′ART}(; kw...)
 NStepBatchSampler{names}(; n, γ, batch_size=32, stack_size=nothing, rng=Random.GLOBAL_RNG) where {names} = NStepBatchSampler{names}(n, γ, batch_size, stack_size, rng)
 
 function sample(s::NStepBatchSampler{names}, ts) where {names}
diff --git a/test/samplers.jl b/test/samplers.jl
@@ -106,7 +106,7 @@ end
     @test size(xs2.terminal) == (batch_size,)
 
     inds = [3, 5, 7]
-    xs3 = RLTrajectories.sample(s1, t2, Val(SSART), inds)
+    xs3 = RLTrajectories.sample(s1, t2, Val(SS′ART), inds)
 
     @test xs3.state == cat(
         (

Original file line number	Diff line number	Diff line change
`@@ -106,7 +106,7 @@ end`
`106`	`106`	`@test size(xs2.terminal) == (batch_size,)`
`107`	`107`
`108`	`108`	`inds = [3, 5, 7]`
`109`		`- xs3 = RLTrajectories.sample(s1, t2, Val(SSART), inds)`
	`109`	`+ xs3 = RLTrajectories.sample(s1, t2, Val(SS′ART), inds)`
`110`	`110`
`111`	`111`	`@test xs3.state == cat(`
`112`	`112`	`(`