Always return heatmap batches as Vector{Matrix{RGB}}

Julia-XAI · Sep 29, 2023 · 1af50fc · 1af50fc
1 parent eea546c
commit 1af50fc
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/heatmap.jl b/src/heatmap.jl
@@ -61,7 +61,7 @@ function heatmap(
     end
     if process_batch
         hs = _heatmap(val, cs, reduce, rangescale, permute)
-        return eachslice(hs; dims=3)
+        return [hs[:, :, i] for i in axes(hs, 3)]
     end
     return [_heatmap(v, cs, reduce, rangescale, permute) for v in eachslice(val; dims=4)]
 end

diff --git a/test/references/heatmaps/process_batch_true.txt b/test/references/heatmaps/process_batch_true.txt
@@ -1 +1 @@
-SubArray{ColorTypes.RGB{Float64}, 2, Array{ColorTypes.RGB{Float64}, 3}, Tuple{Base.Slice{Base.OneTo{Int64}}, Base.Slice{Base.OneTo{Int64}}, Int64}, true}[[RGB{Float64}(1.0,0.5714285714285714,0.5714285714285714) RGB{Float64}(1.0,0.4285714285714283,0.4285714285714283); RGB{Float64}(1.0,0.2857142857142856,0.2857142857142856) RGB{Float64}(1.0,0.1428571428571428,0.1428571428571428)], [RGB{Float64}(0.7142857142857142,0.0,0.0) RGB{Float64}(0.6428571428571428,0.0,0.0); RGB{Float64}(0.5714285714285714,0.0,0.0) RGB{Float64}(0.5,0.0,0.0)]]
+Matrix{ColorTypes.RGB{Float64}}[[RGB{Float64}(1.0,0.5714285714285714,0.5714285714285714) RGB{Float64}(1.0,0.4285714285714283,0.4285714285714283); RGB{Float64}(1.0,0.2857142857142856,0.2857142857142856) RGB{Float64}(1.0,0.1428571428571428,0.1428571428571428)], [RGB{Float64}(0.7142857142857142,0.0,0.0) RGB{Float64}(0.6428571428571428,0.0,0.0); RGB{Float64}(0.5714285714285714,0.0,0.0) RGB{Float64}(0.5,0.0,0.0)]]