Fix ambiguities in similar and blockreshape

Author: mtfishman

Project.toml

@@ -1,7 +1,7 @@
 name = "BlockSparseArrays"
 uuid = "2c9a651f-6452-4ace-a6ac-809f4280fbb4"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.4.1"
+version = "0.4.2"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"

src/BlockArraysExtensions/BlockArraysExtensions.jl

@@ -263,17 +263,20 @@ function blocks_to_cartesianindices(d::Dictionary{<:Block})
   return Dictionary(blocks_to_cartesianindices(eachindex(d)), d)
 end
 
-function blockreshape(a::AbstractArray, dims::Tuple{Vararg{Vector{Int}}})
+function blockreshape(a::AbstractArray, dims::Tuple{Vector{Int},Vararg{Vector{Int}}})
   return blockreshape(a, blockedrange.(dims))
 end
-
-function blockreshape(a::AbstractArray, dims::Vararg{Vector{Int}})
-  return blockreshape(a, dims)
+function blockreshape(a::AbstractArray, dim1::Vector{Int}, dim_rest::Vararg{Vector{Int}})
+  return blockreshape(a, (dim1, dim_rest...))
+end
+# Fix ambiguity error.
+function blockreshape(a::AbstractArray)
+  return blockreshape(a, ())
 end
 
 tuple_oneto(n) = ntuple(identity, n)
 
-function blockreshape(a::AbstractArray, axes::Tuple{Vararg{AbstractUnitRange}})
+function _blockreshape(a::AbstractArray, axes::Tuple{Vararg{AbstractUnitRange}})
   reshaped_blocks_a = reshape(blocks(a), blocklength.(axes))
   reshaped_a = similar(a, axes)
   for I in eachstoredindex(reshaped_blocks_a)
@@ -284,8 +287,20 @@ function blockreshape(a::AbstractArray, axes::Tuple{Vararg{AbstractUnitRange}})
   return reshaped_a
 end
 
-function blockreshape(a::AbstractArray, axes::Vararg{AbstractUnitRange})
-  return blockreshape(a, axes)
+function blockreshape(
+  a::AbstractArray, axes::Tuple{AbstractUnitRange,Vararg{AbstractUnitRange}}
+)
+  return _blockreshape(a, axes)
+end
+# Fix ambiguity error.
+function blockreshape(a::AbstractArray, axes::Tuple{})
+  return _blockreshape(a, axes)
+end
+
+function blockreshape(
+  a::AbstractArray, axis1::AbstractUnitRange, axes_rest::Vararg{AbstractUnitRange}
+)
+  return blockreshape(a, (axis1, axes_rest...))
 end
 
 function cartesianindices(axes::Tuple, b::Block)

src/abstractblocksparsearray/wrappedabstractblocksparsearray.jl

@@ -232,6 +232,18 @@ end
 )
   return blocksparse_similar(a, elt, axes)
 end
+# Fix ambiguity error when non-blocked ranges are passed.
+@interface ::AbstractBlockSparseArrayInterface function Base.similar(
+  a::AbstractArray, elt::Type, axes::Tuple{Base.OneTo,Vararg{Base.OneTo}}
+)
+  return blocksparse_similar(a, elt, axes)
+end
+# Fix ambiguity error when empty axes are passed.
+@interface ::AbstractBlockSparseArrayInterface function Base.similar(
+  a::AbstractArray, elt::Type, axes::Tuple{}
+)
+  return blocksparse_similar(a, elt, axes)
+end
 @interface ::AbstractBlockSparseArrayInterface function Base.similar(
   a::Type{<:AbstractArray}, elt::Type, axes::Tuple{Vararg{Int}}
 )