Fix stats functions signatures (#22)

Author: theogf

src/array_of_similar_arrays.jl

@@ -415,26 +415,26 @@ Statistics.mean(X::AbstractVectorOfSimilarArrays{T,M}) where {T,M} =
 
 
 """
-    var(X::AbstractVectorOfSimilarArrays; corrected::Bool = true)
-    var(X::AbstractVectorOfSimilarArrays, w::StatsBase.AbstractWeights; corrected::Bool = true)
+    var(X::AbstractVectorOfSimilarArrays; corrected::Bool = true, mean = nothing)
+    var(X::AbstractVectorOfSimilarArrays, w::StatsBase.AbstractWeights; corrected::Bool = true, mean = nothing)
 
 Compute the sample variance of the elements vectors of `X`. Equivalent to
 `var` of `flatview(X)` along the last dimension.
 """
-Statistics.var(X::AbstractVectorOfSimilarArrays{T,M}; corrected::Bool = true) where {T,M} =
-    var(flatview(X); dims = M + 1, corrected = corrected)[_ncolons(Val{M}())...]
+Statistics.var(X::AbstractVectorOfSimilarArrays{T,M}; corrected::Bool = true, mean = nothing) where {T,M} =
+    var(flatview(X); dims = M + 1, corrected = corrected, mean = mean)[_ncolons(Val{M}())...]
 
 
 """
-    var(X::AbstractVectorOfSimilarArrays; corrected::Bool = true)
-    var(X::AbstractVectorOfSimilarArrays, w::StatsBase.AbstractWeights; corrected::Bool = true)
+    var(X::AbstractVectorOfSimilarArrays; corrected::Bool = true, mean = nothing)
+    var(X::AbstractVectorOfSimilarArrays, w::StatsBase.AbstractWeights; corrected::Bool = true, mean = nothing)
 
 Compute the sample standard deviation of the elements vectors of `X`.
 Compute the sample variance of the elements vectors of `X`. Equivalent to
 `std` of `flatview(X)` along the last dimension.
 """
-Statistics.std(X::AbstractVectorOfSimilarArrays{T,M}; corrected::Bool = true) where {T,M} =
-    std(flatview(X); dims = M + 1, corrected = corrected)[_ncolons(Val{M}())...]
+Statistics.std(X::AbstractVectorOfSimilarArrays{T,M}; corrected::Bool = true, mean = nothing) where {T,M} =
+    std(flatview(X); dims = M + 1, corrected = corrected, mean = mean)[_ncolons(Val{M}())...]
 
 
 """

src/statsbase_support.jl

@@ -7,11 +7,11 @@ Base.sum(X::AbstractVectorOfSimilarArrays{T,M}, w::StatsBase.AbstractWeights) wh
 Statistics.mean(X::AbstractVectorOfSimilarArrays{T,M}, w::StatsBase.AbstractWeights) where {T,M} =
     mean(flatview(X), w, dims = M + 1)[_ncolons(Val{M}())...]
 
-Statistics.var(X::AbstractVectorOfSimilarArrays{T,M}, w::StatsBase.AbstractWeights; corrected::Bool = true) where {T,M} =
-    var(flatview(X), w, M + 1; corrected = corrected)[_ncolons(Val{M}())...]
+Statistics.var(X::AbstractVectorOfSimilarArrays{T,M}, w::StatsBase.AbstractWeights; mean = nothing, corrected::Bool = true) where {T,M} =
+    var(flatview(X), w, M + 1; mean = mean, corrected = corrected)[_ncolons(Val{M}())...]
 
-Statistics.std(X::AbstractVectorOfSimilarArrays{T,M}, w::StatsBase.AbstractWeights; corrected::Bool = true) where {T,M} =
-    std(flatview(X), w, M + 1; corrected = corrected)[_ncolons(Val{M}())...]
+Statistics.std(X::AbstractVectorOfSimilarArrays{T,M}, w::StatsBase.AbstractWeights; mean = nothing, corrected::Bool = true) where {T,M} =
+    std(flatview(X), w, M + 1; mean = mean, corrected = corrected)[_ncolons(Val{M}())...]
 
 Statistics.cov(X::AbstractVectorOfSimilarVectors, w::StatsBase.AbstractWeights; corrected::Bool = true) =
     cov(flatview(X), w, 2; corrected = corrected)