From 3cbe1cea4723fd6b6c1442e6b5ffa2491471b51b Mon Sep 17 00:00:00 2001 From: Joey Date: Thu, 8 Aug 2024 13:45:28 -0400 Subject: [PATCH] Bug fix --- src/caches/beliefpropagationcache.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/caches/beliefpropagationcache.jl b/src/caches/beliefpropagationcache.jl index bd1f0665..7dc78413 100644 --- a/src/caches/beliefpropagationcache.jl +++ b/src/caches/beliefpropagationcache.jl @@ -16,11 +16,11 @@ using NDTensors: NDTensors default_message(elt, inds_e) = ITensor[denseblocks(delta(elt, i)) for i in inds_e] default_messages(ptn::PartitionedGraph) = Dictionary() -default_message_norm(m::ITensor) = norm(m) function default_message_update(contract_list::Vector{ITensor}; normalize=true, kwargs...) sequence = optimal_contraction_sequence(contract_list) updated_messages = contract(contract_list; sequence, kwargs...) - if normalize && !iszero(norm(updated_messages)) + message_norm = norm(updated_messages) + if normalize && !iszero(message_norm) updated_messages /= message_norm end return ITensor[updated_messages]