Added custom edge specification on vidal_itn_isometries

ITensor · Oct 23, 2023 · 1cf1daf · 1cf1daf
1 parent 8efe301
commit 1cf1daf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/gauging.jl b/src/gauging.jl
@@ -173,10 +173,10 @@ function symmetric_to_vidal_gauge(
 end
 
 """Function to measure the 'isometries' of a state in the Vidal Gauge"""
-function vidal_itn_isometries(ψ::ITensorNetwork, bond_tensors::DataGraph)
+function vidal_itn_isometries(ψ::ITensorNetwork, bond_tensors::DataGraph; edges = vcat(edges(ψ), reverse.(edges(ψ))))
   isometries = DataGraph{vertextype(ψ),ITensor,ITensor}(directed_graph(underlying_graph(ψ)))
 
-  for e in vcat(edges(ψ), reverse.(edges(ψ)))
+  for e in edges
     vsrc, vdst = src(e), dst(e)
     ψv = copy(ψ[vsrc])
     for vn in setdiff(neighbors(ψ, vsrc), [vdst])