From 37533420797ce876aec9b97f09f41f5555bbbd1e Mon Sep 17 00:00:00 2001 From: Bagaev Dmitry Date: Thu, 4 Apr 2024 10:52:04 +0200 Subject: [PATCH] fix BIFM helper --- src/nodes/predefined/bifm_helper.jl | 13 ++++++------- 1 file changed, 6 insertions(+), 7 deletions(-) diff --git a/src/nodes/predefined/bifm_helper.jl b/src/nodes/predefined/bifm_helper.jl index 06a7613ff..e6a65c1d6 100644 --- a/src/nodes/predefined/bifm_helper.jl +++ b/src/nodes/predefined/bifm_helper.jl @@ -21,14 +21,13 @@ struct BIFMHelper <: AbstractFactorNode end @node BIFMHelper Stochastic [out, in] # specify custom functional dependencies for BIFMHelper node -function functional_dependencies(dependencies, factornode::FactorNode{<:Type{BIFMHelper}}, iindex::Int) - cindex = clusterindex(factornode, iindex) +function functional_dependencies(::DefaultFunctionalDependencies, factornode::FactorNode{<:Type{BIFMHelper}}, interface, iindex) + + clusters = getlocalclusters(factornode) + cindex = clusterindex(clusters, iindex) - nodeinterfaces = interfaces(factornode) - nodeclusters = factorisation(factornode) - nodelocalmarginals = localmarginals(factornode) - - varcluster = @inbounds nodeclusters[cindex] + nodeinterfaces = getinterfaces(factornode) + nodelocalmarginals = getmarginals(clusters) # output if iindex === 2