Skip to content

Commit 9739597

Browse files
committed
fix: type issue in ConcatenatedBatch
1 parent af555b1 commit 9739597

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

src/batchs.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,7 +21,7 @@ end
2121
function ConcatenatedBatch((; field)::Batch)
2222
ConcatenatedBatch(cat(field; dims=ndims(field)), vcat([0], field .|> size .|> last |> cumsum))
2323
end
24-
function stack_ConcatenatedBatch(x::AbstractVector{ConcatenatedBatch})
24+
function stack_ConcatenatedBatch(x::AbstractVector{<:ConcatenatedBatch})
2525
field = cat(getfield.(x, :field), dims=ndims(first(x).field))
2626
offsets = vcat([0], getfield.(x, :lengths) .|> last)::Vector{Int} |> cumsum
2727
lengths = zip(getfield.(x, :lengths), offsets) |> Map((lengths, offset) -> lengths .+ offsets) |> vcat

0 commit comments

Comments
 (0)