Fix ReshapedArray wrapper (#11)

ITensor · Dec 22, 2024 · 13b38d0 · 13b38d0
1 parent e112a01
commit 13b38d0
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "SparseArraysBase"
 uuid = "0d5efcca-f356-4864-8770-e1ed8d78f208"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.2.1"
+version = "0.2.2"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"

diff --git a/src/wrappers.jl b/src/wrappers.jl
@@ -58,10 +58,16 @@ function parentindex_to_index(a::PermutedDimsArray{<:Any,N}, I::CartesianIndex{N
 end
 
 using Base: ReshapedArray
-function parentindex_to_index(a::ReshapedArray{<:Any,N}, I::CartesianIndex{N}) where {N}
+# Don't constrain the number of dimensions of the array
+# and index since the parent array can have a different
+# number of dimensions than the `SubArray`.
+function parentindex_to_index(a::ReshapedArray, I::CartesianIndex)
   return CartesianIndices(size(a))[LinearIndices(parent(a))[I]]
 end
-function index_to_parentindex(a::ReshapedArray{<:Any,N}, I::CartesianIndex{N}) where {N}
+# Don't constrain the number of dimensions of the array
+# and index since the parent array can have a different
+# number of dimensions than the `SubArray`.
+function index_to_parentindex(a::ReshapedArray, I::CartesianIndex)
   return CartesianIndices(parent(a))[LinearIndices(size(a))[I]]
 end