Nested ->

Author: wouterwln

src/plugins/meta/meta_macro.jl

@@ -2,6 +2,7 @@ export @meta
 using MacroTools
 
 check_for_returns_meta = (x) -> check_for_returns(x; tag = "meta")
+not_enter_call_walk = guarded_walk((x) -> x isa Expr && x.head == :call)
 
 """
     add_meta_construction(e::Expr)
@@ -82,7 +83,6 @@ Converts all variable references on the left hand side of a meta specification t
 
 # Examples
 """
-
 function convert_meta_variables(e::Expr)
     if @capture(e, (fform_(vars__) -> meta_obj_))
         vars = map(var -> __convert_to_indexed_statement(var), vars)
@@ -129,8 +129,10 @@ function convert_meta_object(e::Expr)
     end
 end
 
+what_walk(::typeof(convert_meta_object)) = not_enter_call_walk
+
 function meta_macro_interior(meta_body::Expr)
-    meta_body = apply_pipeline(meta_body, (x) -> check_for_returns(x; tag = "meta"))
+    meta_body = apply_pipeline(meta_body, check_for_returns_meta)
     meta_body = add_meta_construction(meta_body)
     meta_body = apply_pipeline(meta_body, create_submodel_meta)
     meta_body = apply_pipeline(meta_body, convert_meta_variables)

test/plugins/meta/meta_macro_tests.jl

@@ -299,6 +299,15 @@ end
         push!(__meta__, GraphPPL.MetaObject(GraphPPL.VariableMetaDescriptor(GraphPPL.IndexedVariable(:x, nothing)), some_meta()))
     end
     @test_expression_generating apply_pipeline(input, convert_meta_object) output
+
+    # Test 3: convert_meta_object with  ->
+    input = quote
+        ContinuousTransition() -> CTMeta((H) -> reshape(H, dim, dim))
+    end
+    output = quote
+        push!(__meta__, GraphPPL.MetaObject(GraphPPL.FactorMetaDescriptor(ContinuousTransition, ()), CTMeta((H -> reshape(H, dim, dim)))))
+    end
+    @test_expression_generating apply_pipeline(input, convert_meta_object) output
 end
 
 @testitem "meta_macro_interior" begin