fix for CT node

ReactiveBayes · Apr 4, 2024 · 0f56185 · 0f56185
1 parent 1b64e1c
commit 0f56185
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/nodes/dependencies.jl b/src/nodes/dependencies.jl
@@ -186,7 +186,7 @@ RequireMarginalFunctionalDependencies(μ = vague(NormalMeanPrecision),     τ =
 
 See also: [`ReactiveMP.DefaultFunctionalDependencies`](@ref), [`ReactiveMP.RequireMessageFunctionalDependencies`](@ref), [`ReactiveMP.RequireEverythingFunctionalDependencies`](@ref)
 """
-struct RequireMarginalFunctionalDependencies{S <: NamedTuple}
+struct RequireMarginalFunctionalDependencies{S <: NamedTuple} <: FunctionalDependencies
     specification::S
 end
 
@@ -238,7 +238,7 @@ This includes all inbound messages (including on the same edge) and marginals ov
 
 See also: [`DefaultFunctionalDependencies`](@ref), [`RequireMessageFunctionalDependencies`](@ref), [`RequireMarginalFunctionalDependencies`](@ref)
 """
-struct RequireEverythingFunctionalDependencies end
+struct RequireEverythingFunctionalDependencies <: FunctionalDependencies end
 
 function functional_dependencies(::RequireEverythingFunctionalDependencies, factornode, interface, iindex)
     clusters = getlocalclusters(factornode)

diff --git a/src/nodes/predefined/continuous_transition.jl b/src/nodes/predefined/continuous_transition.jl
@@ -75,7 +75,7 @@ process_Fs(f::Function, a) = [ForwardDiff.jacobian(a -> f(a)[i, :], a) for i in
 
 default_meta(::Type{CTMeta}) = error("ContinuousTransition node requires meta flag explicitly specified")
 
-default_functional_dependencies(::Type{<:ContinuousTransition}) = RequireMarginalFunctionalDependencies((3,), (nothing,))
+default_functional_dependencies(::Type{<:ContinuousTransition}) = RequireMarginalFunctionalDependencies(a = nothing)
 
 """
     `ctcompanion_matrix` casts a vector `a` into a matrix `A` by means of linearization of the transformation function `f` around the expansion point `a0`.