Update NDTensors/src/lib/BlockSparseArrays/src/BlockArraysExtensions/…

…BlockArraysExtensions.jl Co-authored-by: Matt Fishman <[email protected]>
ITensor · Nov 7, 2024 · 780f321 · 780f321
1 parent 85fce19
commit 780f321
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/NDTensors/src/lib/BlockSparseArrays/src/BlockArraysExtensions/BlockArraysExtensions.jl b/NDTensors/src/lib/BlockSparseArrays/src/BlockArraysExtensions/BlockArraysExtensions.jl
@@ -593,7 +593,7 @@ function svd!(A::BlockedMatrix; full::Bool=false, alg::Algorithm=default_svd_alg
 
   # restore block pattern
   m = length(F.S)
-  bsz1, bsz2, bsz3 = BlockArrays.blocksizes(A, 1), [m], BlockArrays.blocksizes(A, 2)
+  bax1, bax2, bax3 = axes(A, 1), blockedrange([m]), axes(A, 2)
 
   u = BlockedArray(F.U, bsz1, bsz2)
   s = BlockedVector(F.S, bsz2)