Cleanup weighted sum (no longer imported from Base)

JuliaStats · Sep 25, 2021 · 1277ff9 · 1277ff9
1 parent e9728b2
commit 1277ff9
Show file tree

Hide file tree

Showing 3 changed files with 233 additions and 262 deletions.
diff --git a/src/Statistics.jl b/src/Statistics.jl
@@ -151,7 +151,7 @@ function _mean!(R::AbstractArray, A::AbstractArray, weights::Nothing)
 end
 
 _mean!(R::AbstractArray, A::AbstractArray, w::AbstractArray) =
-    rmul!(_sum!(R, A, weights=w), inv(sum(w)))
+    rmul!(wsum!(R, A, weights=w), inv(sum(w)))
 
 """
     mean(A::AbstractArray; [dims], [weights::AbstractArray])
@@ -223,7 +223,7 @@ end
 
 # Note: weighted mean currently does not use _mean_promote to avoid overflow
 _mean(::typeof(identity), A::AbstractArray, dims::Colon, w::AbstractArray) =
-    _sum(A, weights=w) / sum(w)
+    wsum(A, weights=w) / sum(w)
 
 _mean(::typeof(identity), A::AbstractArray, dims, w::AbstractArray) =
     _mean!(Base.reducedim_init(t -> (t*zero(eltype(w)))/2, Base.add_sum, A, dims), A, w)