Update src/caches/abstractbeliefpropagationcache.jl

JoeyT1994 · mtfishman · web-flow · commit 29ff79599677 · 2025-04-22T09:35:55.000-04:00
Co-authored-by: Matt Fishman &lt;mtfishman@users.noreply.github.com&gt;
diff --git a/src/caches/abstractbeliefpropagationcache.jl b/src/caches/abstractbeliefpropagationcache.jl
@@ -284,7 +284,7 @@ function update(
   return update(Algorithm(alg), bpc; kwargs...)
 end
 
-function rescale_messages(bp_cache::AbstractBeliefPropagationCache, partitionedge)
+function rescale_messages(bp_cache::AbstractBeliefPropagationCache, partitionedge::PartitionEdge)
   return rescale_messages(bp_cache, [partitionedge])
 end
 
