Merge pull request #286 from ReactiveBayes/273-where-expression-in-the-meta-is-broken

Fix where bug in meta

Author: bvdmitri

src/plugins/meta/meta_macro.jl

@@ -34,6 +34,22 @@ function add_meta_construction(e::Expr)
                 return __meta__
             end
         end
+    elseif @capture(
+        e, (function m_name_(m_args__; m_kwargs__) where {m_where_args__}
+            c_body_
+        end) | (function m_name_(m_args__) where {m_where_args__}
+            c_body_
+        end)
+    )
+        m_kwargs = m_kwargs === nothing ? [] : m_kwargs
+        return quote
+            $(c_body_string_symbol)::String = $(c_body_string)
+            function $m_name($(m_args...); $(m_kwargs...)) where {$(m_where_args...)}
+                __meta__ = GraphPPL.MetaSpecification($(c_body_string_symbol))
+                $c_body
+                return __meta__
+            end
+        end
     else
         return quote
             $(c_body_string_symbol)::String = $(c_body_string)

test/plugins/meta/meta_macro_tests.jl

@@ -386,3 +386,19 @@ end
     end
     @test_expression_generating meta_macro_interior(input) output
 end
+
+@testitem "meta_macro_interior_with_where" begin
+    import GraphPPL: meta_macro_interior
+
+    include("../../testutils.jl")
+
+    @meta function make_learning_meta(f::F) where {F}
+        ContinuousTransition() -> CTMeta(f)
+    end
+
+    @meta function make_learning_meta(f::F, b::G; a::G) where {F, G}
+        ContinuousTransition() -> CTMeta(f)
+    end
+
+    @test length(methods(make_learning_meta)) == 2
+end