Skip to content

Commit f59a058

Browse files
committed
allow dispatch on axes type in sizes_to_blocks
1 parent 0776c2e commit f59a058

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/blockarray.jl

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -233,7 +233,9 @@ mortar(blocks::AbstractArray{R, N}, block_sizes::Vararg{AbstractVector{Int}, N})
233233

234234
mortar(blocks::AbstractArray) = mortar(blocks, sizes_from_blocks(blocks))
235235

236-
function sizes_from_blocks(blocks::AbstractArray{<:Any, N}) where N
236+
sizes_from_blocks(blocks) = sizes_from_blocks(blocks, axes(blocks)) # allow overriding on axes
237+
238+
function sizes_from_blocks(blocks::AbstractArray{<:Any, N}, _) where N
237239
if length(blocks) == 0
238240
return zeros.(Int, size(blocks))
239241
end

0 commit comments

Comments
 (0)