Fix Statistics.mean and var for AbstractVectorOfSimilarArrays

Author: oschulz

src/statsbase_support.jl

@@ -4,10 +4,10 @@
 Base.sum(X::AbstractVectorOfSimilarArrays{T,M}, w::StatsBase.AbstractWeights) where {T,M} = sum(flatview(X), w, M + 1)
 
 Statistics.mean(X::AbstractVectorOfSimilarArrays{T,M}, w::StatsBase.AbstractWeights) where {T,M} =
-    mean(flatview(X), w, dims = M + 1)
+    vec(mean(flatview(X), w, dims = M + 1))
 
 Statistics.var(X::AbstractVectorOfSimilarArrays{T,M}, w::StatsBase.AbstractWeights; corrected::Bool = true) where {T,M} =
-    var(flatview(X), w, M + 1; corrected = corrected)
+    vec(var(flatview(X), w, M + 1; corrected = corrected))
 
 Statistics.cov(X::AbstractVectorOfSimilarVectors, w::StatsBase.AbstractWeights; corrected::Bool = true) =
     cov(flatview(X), w, 2; corrected = corrected)