Skip to content

Commit 367310f

Browse files
committed
collect should be ok on trajectories
1 parent 7f247cc commit 367310f

File tree

3 files changed

+5
-5
lines changed

3 files changed

+5
-5
lines changed

src/controllers.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
export InsertSampleRatioController, InsertSampleController, AsyncInsertSampleRatioController
1+
export InsertSampleRatioController, AsyncInsertSampleRatioController
22

33
"""
44
InsertSampleRatioController(;ratio=1., threshold=1)

test/samplers.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,7 @@ end
3535

3636
append!(t, (a=rand(Int, 10), b=rand(Bool, 10)))
3737

38-
batches = [x for x in t]
38+
batches = collect(t)
3939

4040
@test length(batches) == 10
4141
@test length(batches[1][:policy][:a]) == 3 && length(batches[1][:critic][:b]) == 5
@@ -52,7 +52,7 @@ end
5252

5353
append!(t, (a=rand(Int, 10), b=rand(Bool, 10)))
5454

55-
batches = [x for x in t]
55+
batches = collect(t)
5656

5757
@test length(batches) == 10
5858
@test length(batches[1][:policy][:a]) == 3

test/trajectories.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,11 +6,11 @@
66
),
77
sampler=BatchSampler(3),
88
)
9-
batches = [x for x in t]
9+
batches = collect(t)
1010
@test length(batches) == 0
1111

1212
push!(t, (a=1, b=false))
13-
batches = [x for x in t]
13+
batches = collect(t)
1414
@test length(batches) == 1
1515
end
1616

0 commit comments

Comments
 (0)