Bug fix

Author: JoeyT1994

src/formnetworks/abstractformnetwork.jl

@@ -20,6 +20,13 @@ function SimilarType.similar_type(f::AbstractFormNetwork)
   return typeof(tensornetwork(f))
 end
 
+# TODO: Use `NamedGraphs.GraphsExtensions.parent_graph_type`.
+function data_graph_type(G::Type{<:AbstractFormNetwork})
+  return data_graph_type(fieldtype(G, :tensornetwork))
+end
+# TODO: Use `NamedGraphs.GraphsExtensions.parent_graph`.
+data_graph(f::AbstractFormNetwork) = data_graph(tensornetwork(f))
+
 function operator_vertices(f::AbstractFormNetwork)
   return filter(v -> last(v) == operator_vertex_suffix(f), vertices(f))
 end

src/formnetworks/bilinearformnetwork.jl

@@ -42,10 +42,6 @@ bra_vertex_suffix(blf::BilinearFormNetwork) = blf.bra_vertex_suffix
 ket_vertex_suffix(blf::BilinearFormNetwork) = blf.ket_vertex_suffix
 # TODO: Use `NamedGraphs.GraphsExtensions.parent_graph`.
 tensornetwork(blf::BilinearFormNetwork) = blf.tensornetwork
-# TODO: Use `NamedGraphs.GraphsExtensions.parent_graph_type`.
-data_graph_type(::Type{<:BilinearFormNetwork}) = data_graph_type(tensornetwork(blf))
-# TODO: Use `NamedGraphs.GraphsExtensions.parent_graph`.
-data_graph(blf::BilinearFormNetwork) = data_graph(tensornetwork(blf))
 
 function Base.copy(blf::BilinearFormNetwork)
   return BilinearFormNetwork(

src/formnetworks/linearformnetwork.jl

@@ -36,10 +36,6 @@ bra_vertex_suffix(lf::LinearFormNetwork) = lf.bra_vertex_suffix
 ket_vertex_suffix(lf::LinearFormNetwork) = lf.ket_vertex_suffix
 # TODO: Use `NamedGraphs.GraphsExtensions.parent_graph`.
 tensornetwork(lf::LinearFormNetwork) = lf.tensornetwork
-# TODO: Use `NamedGraphs.GraphsExtensions.parent_graph_type`.
-data_graph_type(::Type{<:LinearFormNetwork}) = data_graph_type(tensornetwork(lf))
-# TODO: Use `NamedGraphs.GraphsExtensions.parent_graph`.
-data_graph(lf::LinearFormNetwork) = data_graph(tensornetwork(lf))
 
 function Base.copy(lf::LinearFormNetwork)
   return LinearFormNetwork(

test/test_itensornetwork.jl

@@ -30,6 +30,7 @@ using ITensors:
   itensor,
   onehot,
   order,
+  prime,
   random_itensor,
   scalartype,
   sim,
@@ -55,7 +56,7 @@ using ITensorNetworks:
   ttn
 using LinearAlgebra: factorize
 using NamedGraphs: NamedEdge
-using NamedGraphs.GraphsExtensions: incident_edges
+using NamedGraphs.GraphsExtensions: disjoint_union, incident_edges
 using NamedGraphs.NamedGraphGenerators: named_comb_tree, named_grid
 using NDTensors: NDTensors, dim
 using Random: randn!
@@ -140,7 +141,7 @@ const elts = (Float32, Float64, Complex{Float32}, Complex{Float64})
     g = named_grid(dims)
     s = siteinds("S=1/2", g)
     ψ = ITensorNetwork(v -> "↑", s)
-    tn = norm_sqr_network(ψ)
+    tn = disjoint_union("bra" => ψ, "ket" => prime(dag(ψ); sites=[]))
     tn_2 = contract(tn, ((1, 2), "ket") => ((1, 2), "bra"))
     @test !has_vertex(tn_2, ((1, 2), "ket"))
     @test tn_2[((1, 2), "bra")] ≈ tn[((1, 2), "ket")] * tn[((1, 2), "bra")]