Use preserve graph

Author: JoeyT1994

Project.toml

@@ -1,7 +1,7 @@
 name = "ITensorNetworks"
 uuid = "2919e153-833c-4bdc-8836-1ea460a35fc7"
 authors = ["Matthew Fishman <[email protected]>, Joseph Tindall <[email protected]> and contributors"]
-version = "0.13.8"
+version = "0.13.9"
 
 [deps]
 AbstractTrees = "1520ce14-60c1-5f80-bbc7-55ef81b5835c"

src/abstractitensornetwork.jl

@@ -607,8 +607,8 @@ function gauge_edge(
   left_inds = uniqueinds(tn, edge)
   ltags = tags(tn, edge)
   X, Y = factorize(tn[src(edge)], left_inds; tags=ltags, ortho="left", kwargs...)
-  tn[src(edge)] = X
-  tn[dst(edge)] *= Y
+  @preserve_graph tn[src(edge)] = X
+  @preserve_graph tn[dst(edge)] = tn[dst(edge)]*Y
   return tn
 end
 
@@ -681,8 +681,8 @@ function _truncate_edge(tn::AbstractITensorNetwork, edge::AbstractEdge; kwargs..
   left_inds = uniqueinds(tn, edge)
   ltags = tags(tn, edge)
   U, S, V = svd(tn[src(edge)], left_inds; lefttags=ltags, kwargs...)
-  tn[src(edge)] = U
-  tn[dst(edge)] *= (S * V)
+  @preserve_graph tn[src(edge)] = U
+  @preserve_graph tn[dst(edge)] = tn[dst(edge)] * (S*V)
   return tn
 end
 

test/test_belief_propagation.jl

@@ -24,14 +24,27 @@ using ITensorNetworks:
   update,
   update_factor,
   updated_message,
-  message_diff
-using ITensors: ITensors, ITensor, combiner, dag, inds, inner, op, prime, random_itensor
+  message_diff,
+  ITensorNetwork
+using ITensors:
+  norm,
+  ITensors,
+  ITensor,
+  @Algorithm_str,
+  Algorithm,
+  combiner,
+  dag,
+  inds,
+  inner,
+  op,
+  prime,
+  random_itensor
 using ITensorNetworks.ModelNetworks: ModelNetworks
 using ITensors.NDTensors: array
 using LinearAlgebra: eigvals, tr
 using NamedGraphs: NamedEdge, NamedGraph
 using NamedGraphs.NamedGraphGenerators: named_comb_tree, named_grid
-using NamedGraphs.PartitionedGraphs: PartitionVertex, partitionedges
+using NamedGraphs.PartitionedGraphs: PartitionVertex, partitionedges, unpartitioned_graph
 using SplitApplyCombine: group
 using StableRNGs: StableRNG
 using TensorOperations: TensorOperations