diff --git a/src/Batches/batch.jl b/src/Batches/batch.jl index d6a8e350..7553ca0d 100644 --- a/src/Batches/batch.jl +++ b/src/Batches/batch.jl @@ -1,4 +1,4 @@ -struct Batch{T,S} <: AbstractVector{T} +struct Batch{T,S} <: ABatch{T} data::CatMat{T,S} end diff --git a/src/Batches/catmat.jl b/src/Batches/catmat.jl index c550f451..9ecc1c9f 100644 --- a/src/Batches/catmat.jl +++ b/src/Batches/catmat.jl @@ -1,6 +1,8 @@ import Base: eltype, size, getindex, setindex!, convert -struct CatMat{T,S} <: AbstractVector{T} +ABatch{T} = AbstractVector{T} + +struct CatMat{T,S} <: ABatch{T} data::S end diff --git a/src/Batches/seq.jl b/src/Batches/seq.jl index 5fd2e1b6..34a453c7 100644 --- a/src/Batches/seq.jl +++ b/src/Batches/seq.jl @@ -1,4 +1,4 @@ -struct Seq{T,S} <: AbstractVector{T} +struct Seq{T,S} <: ABatch{T} data::CatMat{T,S} end