simplify _getvns_link for TypedVarInfo

TuringLang · Oct 8, 2023 · 203aeb3 · 203aeb3
1 parent d15df29
commit 203aeb3
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/varinfo.jl b/src/varinfo.jl
@@ -910,7 +910,7 @@ _getvns_link(varinfo::VarInfo, spl::AbstractSampler) = _getvns(varinfo, spl)
 _getvns_link(varinfo::UntypedVarInfo, spl::SampleFromPrior) = nothing
 _getvns_link(varinfo::TypedVarInfo, spl::SampleFromPrior) = map(
     Base.Returns(nothing),
-    _getvns(varinfo, spl)
+    varinfo.metadata
 )
 
 function link(::DynamicTransformation, varinfo::VarInfo, spl::AbstractSampler, model::Model)