Merge pull request #338 from ReactiveBayes/dev-project-constraint-sup…

…port-ef

Skip projection for ef types

ext/ProjectionExt/ProjectionExt.jl

@@ -14,7 +14,7 @@ function ReactiveMP.prepare_context(constraint::ProjectedTo)
     return ProjectionContext{T}(nothing)
 end
 
-function ReactiveMP.constrain_form(constraint::ProjectedTo, context::ProjectionContext, something::Distribution)
+function ReactiveMP.constrain_form(constraint::ProjectedTo, context::ProjectionContext, something::Union{Distribution, ExponentialFamilyDistribution})
     T = ExponentialFamilyProjection.get_projected_to_type(constraint)
     D = ExponentialFamily.exponential_family_typetag(something)
     if T === D

test/ext/ProjectionExt/inference_with_projection_tests.jl

@@ -494,3 +494,31 @@ end
         plot(p1, p2, p3)
     end
 end
+
+@testitem "Projection constraint should skip processing of `ExponentialFamilyDistribution` instances" begin
+    using BayesBase, ExponentialFamily, Distributions, ExponentialFamilyProjection
+
+    struct NodePrior end
+    struct NodeLikelihood end
+
+    @node NodePrior Stochastic [out, in]
+    @node NodeLikelihood Stochastic [out, in]
+
+    @rule NodePrior(:out, Marginalisation) (q_in::Any,) = NodePrior()
+    @rule NodeLikelihood(:in, Marginalisation) (q_out::Any,) = NodeLikelihood()
+
+    BayesBase.prod(::GenericProd, ::NodePrior, ::NodeLikelihood) = convert(ExponentialFamilyDistribution, Beta(1, 1))
+
+    @model function mymodel(y)
+        a ~ NodePrior(1)
+        y ~ NodeLikelihood(a)
+    end
+
+    constraints = @constraints begin
+        q(a)::ProjectedTo(Beta)
+    end
+
+    result = infer(model = mymodel(), data = (y = 1.0,), constraints = constraints)
+
+    @test result.posteriors[:a] == Beta(1, 1)
+end