allow NamedTuples in more places where Tuple is accepted

joshday · Mar 15, 2018 · 1fcbc8e · 1fcbc8e
1 parent 9e30063
commit 1fcbc8e
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/src/OnlineStatsBase.jl b/src/OnlineStatsBase.jl
@@ -18,10 +18,12 @@ default_weight(o::ExactStat) = EqualWeight()
 abstract type StochasticStat{N} <: OnlineStat{N} end
 default_weight(o::StochasticStat) = LearningRate()
 
-function default_weight(t::Tuple)
+function default_weight(t::Union{Tuple, NamedTuples.NamedTuple})
     W = default_weight(first(t))
-    all(default_weight.(t) .== W) ||
-        error("Weight must be specified when defaults differ.  Found: $(name.(default_weight.(t))).")
+    for item in t 
+        default_weight(item) != W && 
+        error("Weight must be specified when defaults differ. Found:")
+    end
     return W
 end