Bug fixes

Author: mhauru

src/varinfo.jl

@@ -1982,10 +1982,15 @@ function from_linked_internal_transform(::VarNamedVector, ::VarName, dist)
 end
 
 function link(vi::TupleVarInfo, model::Model)
-    metadata = map(value -> link(value, model), vi.metadata)
+    metadata = link(vi.metadata, model)
     return VarInfo(metadata, vi.accs)
 end
 
+function link(vnt::VarNamedTuple, model::Model)
+    new_vnt = map(value -> link(value, model), vnt)
+    return new_vnt
+end
+
 function link(metadata::Metadata, model::Model)
     vns = metadata.vns
     cumulative_logjac = zero(LogProbType)

src/varname.jl

@@ -48,7 +48,9 @@ function remove_trailing_index(vn::VarName{sym,Optic}) where {sym,Optic}
     elseif Optic <: Accessors.IndexLens
         VarName{sym}()
     else
-        prefix(remove_trailing_index(unprefix(vn, sym)), sym)
+        AbstractPPL.prefix(
+            remove_trailing_index(AbstractPPL.unprefix(vn, VarName{sym}())), VarName{sym}()
+        )
     end
 end
 
@@ -58,7 +60,7 @@ function split_trailing_index(vn::VarName{sym,Optic}) where {sym,Optic}
     elseif Optic <: Accessors.IndexLens
         (VarName{sym}(), getoptic(vn))
     else
-        (prefix, index) = split_trailing_index(unprefix(vn, VarName{sym}()))
-        (prefix(prefix, sym), index)
+        (pref, index) = split_trailing_index(AbstractPPL.unprefix(vn, VarName{sym}()))
+        (AbstractPPL.prefix(pref, VarName{sym}()), index)
     end
 end