diff --git a/ext/DistributionsExt/DistributionsExt.jl b/ext/DistributionsExt/DistributionsExt.jl index 68963db..7c28ab6 100644 --- a/ext/DistributionsExt/DistributionsExt.jl +++ b/ext/DistributionsExt/DistributionsExt.jl @@ -8,4 +8,4 @@ import MessagePassingRulesBase: sdtype MessagePassingRulesBase.sdtype(::Type{<:Distribution}) = MessagePassingRulesBase.Stochastic() MessagePassingRulesBase.sdtype(::Distribution) = MessagePassingRulesBase.Stochastic() -end \ No newline at end of file +end diff --git a/src/interfaces/factornodes.jl b/src/interfaces/factornodes.jl index ad70ad5..4c47776 100644 --- a/src/interfaces/factornodes.jl +++ b/src/interfaces/factornodes.jl @@ -274,8 +274,9 @@ function generate_node_expression(node_fform, node_type, node_interfaces) :(MessagePassingRulesBase.collect_factorisation(::$dispatch_type, factorisation::Tuple) = factorisation) else :( - MessagePassingRulesBase.collect_factorisation(::$dispatch_type, factorisation::Tuple) = - ($(ntuple(identity, length(interfaces))),) + MessagePassingRulesBase.collect_factorisation(::$dispatch_type, factorisation::Tuple) = ( + $(ntuple(identity, length(interfaces))), + ) ) end @@ -327,8 +328,7 @@ function generate_node_expression(node_fform, node_type, node_interfaces) # Define the necessary function types result = quote - @doc $doc MessagePassingRulesBase.is_predefined_node(::$dispatch_type) = - MessagePassingRulesBase.PredefinedNodeFunctionalForm() + @doc $doc MessagePassingRulesBase.is_predefined_node(::$dispatch_type) = MessagePassingRulesBase.PredefinedNodeFunctionalForm() MessagePassingRulesBase.sdtype(::$dispatch_type) = (MessagePassingRulesBase.$node_type)() MessagePassingRulesBase.interfaces(::$dispatch_type) = Val($(Tuple(map(first, interfaces)))) @@ -383,4 +383,4 @@ See `?is_predefined_node`. """ macro node(node_fform, node_type, node_interfaces) return esc(generate_node_expression(node_fform, node_type, node_interfaces)) -end \ No newline at end of file +end diff --git a/test/interfaces/factornodes_tests.jl b/test/interfaces/factornodes_tests.jl index eedfc4e..3749e37 100644 --- a/test/interfaces/factornodes_tests.jl +++ b/test/interfaces/factornodes_tests.jl @@ -168,8 +168,9 @@ end @node DummyNodeForNodeFunctionAsDistribution Stochastic [out, mean, var] - Distributions.logpdf(node::DummyNodeForNodeFunctionAsDistribution, out) = - Distributions.logpdf(Distributions.Normal(node.mean, node.var), out) + Distributions.logpdf(node::DummyNodeForNodeFunctionAsDistribution, out) = Distributions.logpdf( + Distributions.Normal(node.mean, node.var), out + ) nodefunction = (out, mean, var) -> Distributions.logpdf(Distributions.Normal(mean, var), out) @@ -191,4 +192,4 @@ end ) ≈ nodefunction(out, mean, var) end end -end \ No newline at end of file +end