Accept AbstractMPS for rearrange_siteinds

tensor4all · Jun 22, 2024 · 64ab8cf · 64ab8cf
1 parent 7725122
commit 64ab8cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/util.jl b/src/util.jl
@@ -434,7 +434,7 @@ function _directprod(M1::MPS, Mx::MPS...)::MPS
     end
 end
 
-function rearrange_siteinds(M::MPS, sites::Vector{Vector{Index{T}}})::MPS where {T}
+function rearrange_siteinds(M::AbstractMPS, sites::Vector{Vector{Index{T}}})::MPS where {T}
     sitesold = siteinds(MPO(collect(M)))
 
     Set(Iterators.flatten(sites)) == Set(Iterators.flatten(sitesold)) || error("siteinds do not match $(sites) != $(sitesold)")