Skip to content

Commit ab664e9

Browse files
committed
resolve comment: add parameter of inner Episode in Episodes
1 parent 344b3b6 commit ab664e9

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

src/traces.jl

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -168,18 +168,19 @@ end
168168
169169
A container for multiple [`Episode`](@ref)s. `init` is a parameterness function which return an empty [`Episode`](@ref).
170170
"""
171-
struct Episodes{names,E} <: AbstractTraces{names,E}
171+
struct Episodes{names,E,T} <: AbstractTraces{names,E}
172172
init::Any
173-
episodes::Vector{Episode}
173+
episodes::Vector{T}
174174
inds::Vector{Tuple{Int,Int}}
175175
end
176176

177177
function Episodes(init)
178178
x = init()
179+
T = typeof(x)
179180
@assert x isa Episode
180181
@assert length(x) == 0
181182
names, E = eltype(x).parameters
182-
Episodes{names,E}(init, [x], Tuple{Int,Int}[])
183+
Episodes{names,E,T}(init, [x], Tuple{Int,Int}[])
183184
end
184185

185186
Base.size(e::Episodes) = size(e.inds)

0 commit comments

Comments
 (0)