feat: add CallWithParent to keep track of callable variables

JuliaSymbolics · Sep 13, 2024 · 1fb4338 · 1fb4338
1 parent e1ba803
commit 1fb4338
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/src/variable.jl b/src/variable.jl
@@ -282,8 +282,10 @@ function Base.show(io::IO, c::CallWithMetadata)
     print(io, "⋆")
 end
 
+struct CallWithParent end
+
 function (f::CallWithMetadata)(args...)
-    metadata(unwrap(f.f(map(unwrap, args)...)), metadata(f))
+    setmetadata(metadata(unwrap(f.f(map(unwrap, args)...)), metadata(f)), CallWithParent, f)
 end
 
 function arg_types_from_call_args(call_args)