Update src/beliefpropagation/beliefpropagation.jl

ITensor · Nov 9, 2023 · f183ec7 · f183ec7
1 parent db9a474
commit f183ec7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/beliefpropagation/beliefpropagation.jl b/src/beliefpropagation/beliefpropagation.jl
@@ -135,7 +135,7 @@ function belief_propagation_iteration(
   mts::DataGraph;
   contract_kwargs=(; alg="density_matrix", output_structure=path_graph_structure, maxdim=1),
   compute_norm=false,
-  edges::Union{Vector{Vector{E}},Vector{E}}=belief_propagation_edge_sequence(
+  edges::Union{Vector{Vector{<:NamedEdge}},Vector{<:NamedEdge}}=belief_propagation_edge_sequence(
     undirected_graph(underlying_graph(mts))
   ),
 ) where {E<:NamedEdge}