fix groupby with heatmap plot

Author: joshday

src/OnlineStats.jl

@@ -114,8 +114,8 @@ include("viz/khist.jl")
 include("viz/khist2d.jl")
 include("viz/partition.jl")
 include("viz/mosaicplot.jl")
-include("viz/recipes.jl")
 include("viz/heatmap.jl")
+include("viz/recipes.jl")
 include("viz/hexlattice.jl")
 include("viz/ash.jl")
 end

src/stats/histograms.jl

@@ -186,7 +186,7 @@ counts(o::ExpandingHist) = o.counts
 edges(o::ExpandingHist) = o.edges
 
 function _fit!(o::ExpandingHist, y)
-    isfinite(y) || error("ExpandingHist requires finite values.  Found $y.")
+    isfinite(y) || error("ExpandingHist requires `isfinite` values.  Found $y.")
     o.n += 1
 
     # init

src/viz/recipes.jl

@@ -71,7 +71,7 @@ end
 end
 
 #-----------------------------------------------------------------------# GroupBy
-@recipe function f(o::GroupBy{<:Any, <:Any, <:Union{KHist, ExpandingHist, Hist}})
+@recipe function f(o::GroupBy{<:Any, <:Any, <:Union{KHist, ExpandingHist, Hist, HeatMap}})
     sort!(o.value)
     link --> :all
     for (k, v) in pairs(o.value)
@@ -81,6 +81,19 @@ end
         end
     end
 end
+@recipe function f(o::GroupBy{<:Any, <:Any, <:HeatMap})
+    sort!(o.value)
+    link --> :all
+    layout --> length(o.value)
+    for (i, (k, v)) in enumerate(pairs(o.value))
+        @series begin
+            subplot --> i
+            marginals --> false
+            title --> string(k)
+            v
+        end
+    end
+end
 @recipe function f(o::GroupBy)
     sort!(o.value)
     link --> :all