Update NDTensors/src/lib/BlockSparseArrays/src/BlockArraysExtensions/BlockArraysExtensions.jl

Co-authored-by: Matt Fishman <[email protected]>

Author: lkdvos

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)