Fix Data Graph Type

ITensor · Mar 7, 2024 · 4067de0 · 4067de0
1 parent b11a0f6
commit 4067de0
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/gauging.jl b/src/gauging.jl
@@ -11,7 +11,9 @@ site_tensors(ψ::VidalITensorNetwork) = ψ.itensornetwork
 bond_tensors(ψ::VidalITensorNetwork) = ψ.bond_tensors
 bond_tensor(ψ::VidalITensorNetwork, e) = bond_tensors(ψ)[e]
 
-data_graph_type(TN::Type{VidalITensorNetwork}) = data_graph_type(site_tensors(TN))
+function data_graph_type(TN::Type{<:VidalITensorNetwork})
+  return data_graph_type(fieldtype(TN, :itensornetwork))
+end
 data_graph(ψ::VidalITensorNetwork) = data_graph(site_tensors(ψ))
 function copy(ψ::VidalITensorNetwork)
   return VidalITensorNetwork(copy(site_tensors(ψ)), copy(bond_tensors(ψ)))