fix BIFM helper

ReactiveBayes · Apr 4, 2024 · 3753342 · 3753342
1 parent 4017a99
commit 3753342
Showing 1 changed file with 6 additions and 7 deletions.
diff --git 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